oneword-ai 0.1.0__py3-none-any.whl

oneword_ai-0.1.0.dist-info/METADATA

@@ -0,0 +1,237 @@
+Metadata-Version: 2.4
+Name: oneword-ai
+Version: 0.1.0
+Summary: A minimalist one-word subtitle generator with Neobrutalism UI
+Author-email: Ambrish <ambrishyadav1110@gmail.com>
+License: Copyright 2025 Ambrish
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+Project-URL: Homepage, https://github.com/Ambrishyadav-byte/OnewordAI
+Project-URL: Bug Tracker, https://github.com/Ambrishyadav-byte/OnewordAI/issues
+Keywords: subtitles,ai,whisper,content-creation,video-editing
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: license.txt
+Requires-Dist: openai-whisper>=20231117
+Requires-Dist: torch>=2.0.0
+Requires-Dist: fastapi>=0.100.0
+Requires-Dist: uvicorn[standard]>=0.23.0
+Requires-Dist: python-multipart>=0.0.6
+Requires-Dist: ffmpeg-python>=0.2.0
+Requires-Dist: gradio>=4.0.0
+Requires-Dist: tqdm>=4.65.0
+Requires-Dist: aiofiles>=23.0.0
+Requires-Dist: transformers>=4.30.0
+Requires-Dist: accelerate>=0.20.0
+Requires-Dist: huggingface_hub>=0.16.0
+Dynamic: license-file
+
+# 🎬 OneWord AI - Subtitle Generator
+
+<div align="center">
+
+**Generate cinematic one-word subtitles from video/audio using Whisper AI**
+
+[![Python](https://img.shields.io/badge/Python-3.8%2B-blue)](https://www.python.org/)
+[![License](https://img.shields.io/badge/License-MIT-green)](license.txt)
+[![Whisper](https://img.shields.io/badge/OpenAI-Whisper-orange)](https://github.com/openai/whisper)
+
+Perfect for creators making high-energy reels, shorts, and TikToks!
+
+</div>
+
+---
+
+## ✨ Features
+
+- 🎯 **Three Subtitle Modes**: One Word, Two Word Punch, Phrase Mode
+- 🌍 **Multi-Language**: Auto-detect or specify (English, Hindi, Urdu, Spanish)
+- 🤖 **Multiple Models**: Medium, Large, and **Hindi2Hinglish** (Oriserve/Prime) 🆕
+- 📦 **Python Package**: Installable via pip with `oneword-cli` and `oneword-web` commands
+- 💻 **Local CLI**: Robust command-line tool for batch processing
+- 🌐 **Web UI**: Beautiful Neobrutalism-styled web interface
+- ☁️ **Cloud Ready**: Works on Google Colab and Hugging Face Spaces
+- 🐳 **Docker Support**: Containerized deployment
+
+## 🚀 Quick Start
+
+### Installation
+
+```bash
+# Clone the repository
+git clone https://github.com/Ambrishyadav-byte/OnewordAI.git
+cd OnewordAI
+
+# Install as a package
+pip install -e .
+```
+
+**Prerequisites**: Ensure [FFmpeg](https://ffmpeg.org/) is installed on your system.
+
+### Usage
+
+#### 🖥️ CLI (Command Line)
+
+See [CLI.md](CLI.md) for full documentation.
+
+```bash
+# Basic usage
+oneword-cli -i video.mp4
+
+# Full options
+oneword-cli -i video.mp4 -m medium -lang hi -mode oneword
+```
+
+#### 🌐 Web UI
+
+```bash
+# Start server & open browser
+oneword-web
+```
+
+Features:
+- Drag & drop file upload
+- Real-time progress tracking
+- Instant SRT download
+- Responsive Neobrutalism design
+
+#### ☁️ Google Colab
+
+[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/)
+
+1. Open `OneWord_Colab.ipynb` in Colab
+2. Run all cells
+3. Upload your video
+4. Download your SRT!
+
+#### 🤗 Hugging Face Space
+
+```bash
+python app_gradio.py
+```
+
+Or deploy to Hugging Face Spaces for a hosted version!
+
+## 📊 Subtitle Modes
+
+| Mode | Description | Use Case |
+|------|-------------|----------|
+| **One Word** | Each word = separate subtitle | High-energy, attention-grabbing content |
+| **Two Word Punch** | Groups of 2 words | Punchy, impactful messaging |
+| **Phrase Mode** | Full sentence segments | Traditional subtitle style |
+
+## 🎨 Web UI Preview
+
+The web interface features a stunning **Neobrutalism** design:
+- Bold black borders
+- Vibrant color palette
+- Sharp shadows
+- Grid background pattern
+- Smooth animations
+
+## 🐳 Docker Deployment
+
+```bash
+# Build image
+docker build -t oneword-ai .
+
+# Run container
+docker run -p 8000:8000 oneword-ai
+```
+
+## 📁 Project Structure
+
+```
+minimalist-one-word-subtitle-generator/
+├── onewordai/                    # Source code (package)
+│   ├── core/
+│   │   ├── __init__.py
+│   │   └── engine.py          # Core subtitle generation logic
+│   ├── api/
+│   │   ├── __init__.py
+│   │   └── main.py            # FastAPI backend
+│   └── web/
+│       ├── index.html         # Web UI
+│       ├── style.css          # Neobrutalism styles
+│       └── app.js             # Frontend logic
+├── cli.py                     # CLI interface
+├── app_gradio.py              # Gradio app for HF Spaces
+├── OneWord_Colab.ipynb        # Google Colab notebook
+├── Dockerfile                 # Docker configuration
+├── requirements.txt           # Python dependencies
+└── README.md
+```
+
+## 🛠️ Development
+
+### API Endpoints
+
+- `POST /upload` - Upload video/audio file
+- `POST /process` - Start transcription job
+- `GET /status/{job_id}` - Check job progress
+- `GET /download/{job_id}` - Download generated SRT
+
+### Requirements
+
+- Python 3.8+
+- FFmpeg
+- PyTorch
+- OpenAI Whisper
+- FastAPI (for web server)
+- Gradio (for HF Spaces)
+
+## 💡 Tips for Creators
+
+### Video Editing Workflow
+
+1. Generate SRT using OneWord AI
+2. Import into your editor:
+   - **CapCut**: Text → Local Captions → Upload
+   - **VN Editor**: Text → SRT → Import
+   - **Premiere Pro**: File → Import → Captions
+3. Apply animations (Pop, Spring, Bounce)
+4. Customize colors and fonts
+
+### Best Practices
+
+- Use **Tiny** model for quick drafts
+- Use **Base** model for production (best balance)
+- Use **Small** model for technical/complex content
+- **One Word** mode works best for 30-60 sec reels
+- Enable language selection for multilingual content
+
+## 🤝 Contributing
+
+Contributions welcome! Open an issue or submit a PR.
+
+Ideas for improvements:
+- Auto-capitalization for emphasis
+- Color-coded keywords
+- Export with burned-in subtitles
+- Batch processing multiple files
+
+## 📜 License
+
+MIT License - see [license.txt](license.txt)
+
+## 🤝 Credits
+
+- [OpenAI Whisper](https://github.com/openai/whisper) - Speech recognition
+- [FastAPI](https://fastapi.tiangolo.com/) - Backend framework
+- [Gradio](https://gradio.app/) - ML web interfaces
+
+Built with ❤️ by [Ambrish](https://github.com/ambrish-yadav)
+
+Follow for updates: [@ambrish.yadav.1](https://instagram.com/ambrish.yadav.1)
+
+---
+
+<div align="center">
+
+⭐ Star this repo if you find it useful!
+
+</div>

oneword_ai-0.1.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+oneword_ai-0.1.0.dist-info/licenses/license.txt,sha256=ZlGk1ZMpgK_LQB1VTiN27JbvU-KL31byRxgM7wAmEu4,1060
+onewordai/__init__.py,sha256=DW7ib3o1W0lchP960Mqdg_a-_aPPhPWgYKu5efR56ME,73
+onewordai/cli.py,sha256=kz8epolGelaSfbOXdIB8IaiwUBLaKD8uyP6RBGOkCMM,2008
+onewordai/api/__init__.py,sha256=W5bQNZGz2YLMXx1hC9e_eKhbf_GNFVYy3MC3D7gugUc,20
+onewordai/api/main.py,sha256=QGyg4q72QkodF4HFAz6KOA8d3AlucisUB1OAw2XpANk,7964
+onewordai/core/__init__.py,sha256=oTsOiTuWDAh1O6N9ZXfNFuiCmP3R99dBkEFljt3LI6Q,114
+onewordai/core/engine.py,sha256=CqHVfvRjXSHgkjcPagYEhvCGwM2ywSlhyTvTran7MK0,15631
+onewordai/web/app.js,sha256=mOp_RRear_N8RTpNEsYs2oj6mkji_AaKXlpxFsAtH-M,11166
+onewordai/web/index.html,sha256=r1OSgeb35diHKmj1uWGnd-ltPBPz6aQp5mmcp2jTDmY,5995
+onewordai/web/style.css,sha256=InsF-b_e5-zRhwaUHKf7cHnenNv87zA82VjcjqIsOnE,10485
+oneword_ai-0.1.0.dist-info/METADATA,sha256=v5LBayOqcOffEvF6G7JXLumYGzdocKO1C5pzGy0ZeaE,7621
+oneword_ai-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+oneword_ai-0.1.0.dist-info/entry_points.txt,sha256=qJ4u04rCiYtI5fCyPRBfTzW_-xc05BCSEct8YHzXX9Y,97
+oneword_ai-0.1.0.dist-info/top_level.txt,sha256=otwwf_hwyKfUYUJnculUSznHz9iCFJRrHgIHyLQRbfE,10
+oneword_ai-0.1.0.dist-info/RECORD,,

oneword_ai-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

oneword_ai-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+oneword-cli = onewordai.cli:main
+oneword-web = onewordai.api.main:start_server

oneword_ai-0.1.0.dist-info/licenses/license.txt

@@ -0,0 +1,7 @@
+Copyright 2025 Ambrish
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

oneword_ai-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+onewordai

onewordai/__init__.py

@@ -0,0 +1,2 @@
+"""OneWord AI - Cinematic subtitle generator."""
+__version__ = "2.0.0"

onewordai/api/__init__.py

@@ -0,0 +1 @@
+"""API package."""

onewordai/api/main.py

@@ -0,0 +1,262 @@
+"""
+FastAPI backend for OneWord AI Subtitle Generator.
+Handles file uploads, transcription processing, and SRT downloads.
+"""
+from fastapi import FastAPI, File, UploadFile, Form, HTTPException, BackgroundTasks
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+from pathlib import Path
+import uuid
+import shutil
+import asyncio
+from typing import Optional, Dict
+from datetime import datetime
+
+from onewordai.core.engine import SubtitleGenerator
+
+app = FastAPI(title="OneWord AI Subtitle Generator")
+
+# CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Directories
+UPLOAD_DIR = Path("uploads")
+OUTPUT_DIR = Path("outputs")
+UPLOAD_DIR.mkdir(exist_ok=True)
+OUTPUT_DIR.mkdir(exist_ok=True)
+
+# Job storage (in-memory, replace with Redis for production)
+jobs: Dict[str, dict] = {}
+
+
+class JobStatus:
+    PENDING = "pending"
+    PROCESSING = "processing"
+    COMPLETED = "completed"
+    FAILED = "failed"
+    CANCELLED = "cancelled"
+
+
+def process_video_task(
+    job_id: str,
+    input_path: str,
+    model: str,
+    language: Optional[str],
+    mode: str
+):
+    """Background task to process video."""
+    try:
+        jobs[job_id]["status"] = JobStatus.PROCESSING
+        jobs[job_id]["progress"] = 0
+        
+        # Create generator
+        generator = SubtitleGenerator(model_name=model)
+        
+        # Process
+        output_path = OUTPUT_DIR / f"{job_id}.srt"
+        
+        def progress_callback(percent):
+            jobs[job_id]["progress"] = percent
+            
+        def status_callback(msg):
+            jobs[job_id]["status_message"] = msg
+        
+        generator.process(
+            input_path=input_path,
+            output_path=str(output_path),
+            language=language,
+            mode=mode,
+            progress_callback=progress_callback,
+            status_callback=status_callback
+        )
+        
+        jobs[job_id]["status"] = JobStatus.COMPLETED
+        jobs[job_id]["progress"] = 100
+        jobs[job_id]["output_file"] = str(output_path)
+        jobs[job_id]["completed_at"] = datetime.now().isoformat()
+        
+    except Exception as e:
+        jobs[job_id]["status"] = JobStatus.FAILED
+        jobs[job_id]["error"] = str(e)
+        print(f"❌ Job {job_id} failed: {e}")
+
+
+
+@app.get("/api/health")
+async def health_check():
+    """Health check."""
+    return {"status": "ok", "service": "OneWord AI Subtitle Generator"}
+
+
+@app.post("/api/upload")
+async def upload_file(file: UploadFile = File(...)):
+    """
+    Upload a video/audio file.
+    Returns a file_id for use in processing.
+    """
+    if not file.filename:
+        raise HTTPException(status_code=400, detail="No file provided")
+    
+    # Generate unique file ID
+    file_id = str(uuid.uuid4())
+    file_ext = Path(file.filename).suffix
+    file_path = UPLOAD_DIR / f"{file_id}{file_ext}"
+    
+    # Save file
+    with file_path.open("wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    
+    return {
+        "file_id": file_id,
+        "filename": file.filename,
+        "size": file_path.stat().st_size
+    }
+
+
+@app.post("/api/process")
+async def process_file(
+    background_tasks: BackgroundTasks,
+    file_id: str = Form(...),
+    model: str = Form("medium"),
+    language: Optional[str] = Form(None),
+    mode: str = Form("oneword")
+):
+    """
+    Start processing a previously uploaded file.
+    Returns a job_id to track progress.
+    """
+    # Find uploaded file
+    uploaded_files = list(UPLOAD_DIR.glob(f"{file_id}.*"))
+    if not uploaded_files:
+        raise HTTPException(status_code=404, detail="File not found")
+    
+    input_path = str(uploaded_files[0])
+    
+    # Validate inputs
+    allowed_models = ["medium", "large", "Oriserve/Whisper-Hindi2Hinglish-Prime"]
+    if model not in allowed_models:
+        raise HTTPException(status_code=400, detail="Invalid model")
+    
+    if language and language not in ["hi", "en", "ur", "es"]:
+        raise HTTPException(status_code=400, detail="Invalid language")
+    
+    if mode not in ["oneword", "twoword", "phrase"]:
+        raise HTTPException(status_code=400, detail="Invalid mode")
+    
+    # Create job
+    job_id = str(uuid.uuid4())
+    jobs[job_id] = {
+        "job_id": job_id,
+        "file_id": file_id,
+        "status": JobStatus.PENDING,
+        "progress": 0,
+        "status_message": "📦 Checking model... (First-time download ~1.5GB)" if "/" in model else "🚀 Preparing to transcribe...",
+        "model": model,
+        "language": language,
+        "mode": mode,
+        "created_at": datetime.now().isoformat()
+    }
+    
+    # Start background task
+    background_tasks.add_task(
+        process_video_task,
+        job_id, input_path, model, language, mode
+    )
+    
+    return {"job_id": job_id}
+
+
+@app.get("/api/status/{job_id}")
+async def get_status(job_id: str):
+    """Get processing status for a job."""
+    if job_id not in jobs:
+        raise HTTPException(status_code=404, detail="Job not found")
+    
+    return jobs[job_id]
+
+
+@app.get("/api/download/{job_id}")
+async def download_srt(job_id: str):
+    """Download the generated SRT file."""
+    if job_id not in jobs:
+        raise HTTPException(status_code=404, detail="Job not found")
+    
+    job = jobs[job_id]
+    
+    if job["status"] != JobStatus.COMPLETED:
+        raise HTTPException(status_code=400, detail="Job not completed yet")
+    
+    output_file = Path(job["output_file"])
+    if not output_file.exists():
+        raise HTTPException(status_code=404, detail="Output file not found")
+    
+    return FileResponse(
+        output_file,
+        media_type="application/x-subrip",
+        filename=f"subtitles_{job['mode']}.srt"
+    )
+
+
+@app.post("/api/cancel/{job_id}")
+async def cancel_job(job_id: str):
+    """Cancel a running job."""
+    if job_id not in jobs:
+        raise HTTPException(status_code=404, detail="Job not found")
+    
+    job = jobs[job_id]
+    
+    # Only cancel if still processing
+    if job["status"] in [JobStatus.PENDING, JobStatus.PROCESSING]:
+        jobs[job_id]["status"] = JobStatus.CANCELLED
+        jobs[job_id]["status_message"] = "❌ Cancelled by user"
+        return {"message": "Job cancelled successfully"}
+    else:
+        return {"message": f"Job already {job['status']}"}
+
+
+# Mount static files (web UI) at root
+try:
+    # Try resolving relative to package install location
+    package_root = Path(__file__).parent.parent
+    web_dir = package_root / "web"
+    
+    if not web_dir.exists():
+        # Fallback for local development if running from root
+        web_dir = Path("onewordai/web")
+        
+    app.mount("/", StaticFiles(directory=str(web_dir), html=True), name="web")
+except Exception as e:
+    print(f"⚠️ Warning: Could not mount web UI: {e}")
+
+
+if __name__ == "__main__":
+    import uvicorn
+    print("\n✨ OneWord AI running at: http://localhost:8000\n")
+    uvicorn.run(app, host="0.0.0.0", port=8000)
+
+# Entry point for package
+def start_server():
+    """Start the web server (used by oneword-web command)."""
+    import uvicorn
+    import webbrowser
+    import threading
+    import time
+    
+    def open_browser():
+        time.sleep(2)
+        print("🚀 Opening browser at http://localhost:8000")
+        webbrowser.open("http://localhost:8000")
+    
+    threading.Thread(target=open_browser, daemon=True).start()
+    
+    # Run server
+    print("✨ Starting OneWord AI Web Server...")
+    # Use reload=False for production package usage
+    uvicorn.run("onewordai.api.main:app", host="0.0.0.0", port=8000, reload=False)

onewordai/cli.py

@@ -0,0 +1,67 @@
+"""
+CLI interface for OneWord AI Subtitle Generator.
+"""
+import argparse
+from pathlib import Path
+from .core.engine import SubtitleGenerator
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="OneWord AI - Generate cinematic one-word subtitles using Whisper"
+    )
+    parser.add_argument(
+        "-i", "--input", 
+        required=True, 
+        help="Path to input video/audio file"
+    )
+    parser.add_argument(
+        "-m", "--model", 
+        default="medium",
+        choices=["medium", "large", "Oriserve/Whisper-Hindi2Hinglish-Prime"],        help="Whisper model to use (default: medium)"
+    )
+    parser.add_argument(
+        "-lang", "--language",
+        default=None,
+        choices=["hi", "en", "ur", "es"],
+        help="Language code (hi=Hindi, en=English, ur=Urdu, es=Spanish). Auto-detect if not specified."
+    )
+    parser.add_argument(
+        "-mode", "--mode",
+        default="oneword",
+        choices=["oneword", "twoword", "phrase"],
+        help="Subtitle mode: oneword (default), twoword (punch effect), phrase (full segment)"
+    )
+    parser.add_argument(
+        "-o", "--output",
+        help="Output SRT file path (optional, auto-generated if not specified)"
+    )
+    
+    args = parser.parse_args()
+    
+    # Validate input file
+    input_path = Path(args.input)
+    if not input_path.exists():
+        print(f"❌ Error: Input file '{args.input}' not found.")
+        return 1
+    
+    # Create generator
+    generator = SubtitleGenerator(model_name=args.model)
+    
+    # Process
+    try:
+        output_file = generator.process(
+            input_path=str(input_path),
+            output_path=args.output,
+            language=args.language,
+            mode=args.mode
+        )
+        print(f"\n🎉 Done! SRT file: {output_file}")
+        return 0
+    except Exception as e:
+        print(f"\n❌ Error: {e}")
+        return 1
+
+
+if __name__ == "__main__":
+    exit(main())

onewordai/core/__init__.py

@@ -0,0 +1,4 @@
+"""Core subtitle generation engine."""
+from .engine import SubtitleGenerator
+
+__all__ = ["SubtitleGenerator"]