Create app.py

app.py

@@ -0,0 +1,144 @@
+from fastapi import FastAPI, UploadFile, File, HTTPException, BackgroundTasks, Query
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
+import os
+import uuid
+import time
+from typing import Dict, List, Optional
+import threading
+
+# Import processing functions and variables
+from processing_logic import (
+    processing_status, 
+    uploaded_mp4s, 
+    log_message,
+    process_hf_files_background,
+    UPLOAD_DIRECTORY,
+    MP4_OUTPUT_FOLDER,
+    hf_api,
+    DEFAULT_RAR_LIMIT
+)
+
+app = FastAPI(title="MP4 Processing API", description="API for uploading, processing, and downloading MP4 files")
+
+# Configure CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
+processing_thread = None
+
+# ==== API ENDPOINTS ====
+
+def save_file(uploaded_file: UploadFile, save_path: str):
+    os.makedirs(os.path.dirname(save_path), exist_ok=True)
+    with open(save_path, "wb") as f:
+        shutil.copyfileobj(uploaded_file.file, f)
+
+# === ROUTES ===
+
+@app.get("/")
+def root():
+    return {"message": "FastAPI Florence-2 backend"}
+
+@app.post("/upload/")
+async def upload_file(file: UploadFile = File(...)):
+    """Uploads a file and stores it by filename inside the upload folder"""
+    save_path = os.path.join(MP4_UPLOAD_FOLDER, file.filename)
+    save_file(file, save_path)
+    return {"status": "uploaded", "filename": file.filename}
+
+@app.get("/list")
+async def list_uploaded_files():
+    """List uploaded MP4 files"""
+    files = []
+    if os.path.exists(MP4_UPLOAD_FOLDER):
+        for file in os.listdir(MP4_UPLOAD_FOLDER):
+            if file.lower().endswith(".mp4"):
+                files.append(file)
+    return JSONResponse(content=files)
+
+
+@app.get("/download/{mp4_path:path}")
+async def download_file(mp4_path: str):
+    """
+    Download any MP4 under processed_mp4s, including nested course folders.
+    e.g. GET /download/MDS_CGAdventure_DownloadPirate.com/1.mp4
+    """
+    # Compose the full file path
+    full_path = os.path.join(MP4_OUTPUT_FOLDER, mp4_path)
+
+    # Security: Prevent escaping outside of MP4_OUTPUT_FOLDER
+    full_path = os.path.normpath(full_path)
+    if not full_path.startswith(os.path.abspath(MP4_OUTPUT_FOLDER)):
+        raise HTTPException(status_code=400, detail="Invalid path")
+
+    if not os.path.exists(full_path) or not full_path.lower().endswith(".mp4"):
+        raise HTTPException(status_code=404, detail="File not found")
+
+    return FileResponse(path=full_path, filename=os.path.basename(full_path), media_type="video/mp4")
+
+@app.get("/courses")
+async def list_courses():
+    """List all unique course folder names based on extracted MP4 paths."""
+    course_folders = set()
+
+    if os.path.exists(MP4_OUTPUT_FOLDER):
+        for root, dirs, files in os.walk(MP4_OUTPUT_FOLDER):
+            for file in files:
+                if file.lower().endswith(".mp4"):
+                    rel_path = os.path.relpath(os.path.join(root, file), MP4_OUTPUT_FOLDER)
+                    course_folder = os.path.dirname(rel_path)
+                    if course_folder:
+                        course_folders.add(course_folder)
+
+    return JSONResponse(content={"courses": sorted(course_folders), "total": len(course_folders)})
+
+@app.get("/images/{course_folder:path}")
+async def list_mp4s_in_course(course_folder: str):
+    """List MP4s available for a specific course folder"""
+    course_path = os.path.join(MP4_OUTPUT_FOLDER, course_folder)
+    if not os.path.exists(course_path):
+        raise HTTPException(status_code=404, detail="Course folder not found")
+
+    mp4s = []
+    for file in os.listdir(course_path):
+        if file.lower().endswith(".mp4"):
+            mp4s.append(file)
+
+    return JSONResponse(content=mp4s)
+
+@app.get("/")
+async def root():
+    """API root endpoint"""
+    return {
+        "message": "MP4 Processing API",
+        "version": "1.0.0",
+        "endpoints": {
+            "upload": "POST /upload - Upload MP4 file",
+            "download": "GET /download/{mp4_id} - Download MP4 file",
+            "delete": "DELETE /delete/{mp4_id} - Delete MP4 file",
+            "list": "GET /mp4s - List all MP4 files",
+            "start_processing": "POST /process/start - Start HF processing",
+            "processing_status": "GET /process/status - Get processing status",
+            "stop_processing": "POST /process/stop - Stop processing"
+        }
+    }
+
+@app.on_event("startup")
+async def startup_event():
+    """Run the processing loop in the background when the API starts"""
+    global processing_thread
+    if not (processing_thread and processing_thread.is_alive()):
+        # log_message("🚀 Starting RAR extraction, frame extraction, and vision analysis pipeline in background...")
+        processing_thread = threading.Thread(target=process_hf_files_background)
+        processing_thread.daemon = True
+        processing_thread.start()
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)