Update app.py

app.py

@@ -5,14 +5,21 @@ import subprocess
 import shutil
 import time
 import sys
+import threading
 from typing import Dict, List, Optional, Any
 from huggingface_hub import HfApi, hf_hub_url
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
+import uvicorn
 
 # Fix Unicode encoding for Windows
 if sys.platform == 'win32':
     import io
     sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
 
+# Initialize FastAPI app
+app = FastAPI(title="Audio Transcriber", description="Audio transcription and upload service")
+
 # ==== CONFIGURATION ====
 HF_TOKEN = ""
 SOURCE_REPO_ID = "Fred808/BG3"  # Fetch audio files from here
@@ -532,3 +539,139 @@ def main_processing_loop():
 
 if __name__ == "__main__":
     main_processing_loop()
+
+# ===== FASTAPI ENDPOINTS =====
+
+@app.get("/")
+async def root():
+    """Root endpoint with service info"""
+    return {
+        "service": "Audio Transcriber",
+        "status": "running",
+        "version": "1.0.0",
+        "endpoints": {
+            "status": "/status",
+            "start": "/start",
+            "stop": "/stop",
+            "process": "/process/{filename}",
+            "logs": "/logs"
+        }
+    }
+
+@app.get("/status")
+async def get_status():
+    """Get current processing status"""
+    return {
+        "is_running": processing_status["is_running"],
+        "current_file": processing_status["current_file"],
+        "total_files": processing_status["total_files"],
+        "processed_files": processing_status["processed_files"],
+        "transcribed_files": processing_status["transcribed_files"],
+        "failed_files": processing_status["failed_files"],
+        "last_update": processing_status["last_update"],
+        "recent_logs": processing_status["logs"][-10:]
+    }
+
+@app.post("/start")
+async def start_processing():
+    """Start the main processing loop"""
+    if processing_status["is_running"]:
+        raise HTTPException(status_code=400, detail="Processing already running")
+    
+    # Start processing in a separate thread
+    thread = threading.Thread(target=main_processing_loop, daemon=True)
+    thread.start()
+    
+    return {
+        "message": "Processing started",
+        "status": "started"
+    }
+
+@app.post("/stop")
+async def stop_processing():
+    """Stop the main processing loop"""
+    if not processing_status["is_running"]:
+        raise HTTPException(status_code=400, detail="Processing not running")
+    
+    processing_status["is_running"] = False
+    
+    return {
+        "message": "Processing stopped",
+        "status": "stopped"
+    }
+
+@app.get("/logs")
+async def get_logs(limit: int = 50):
+    """Get recent logs"""
+    logs = processing_status["logs"][-limit:]
+    return {
+        "total_logs": len(processing_status["logs"]),
+        "recent_logs": logs
+    }
+
+@app.post("/process/{filename}")
+async def process_single_file(filename: str):
+    """Process a single audio file manually"""
+    try:
+        log_message(f"🎯 Manual processing requested for: {filename}", "INFO")
+        
+        # Download and process the file
+        reference_map = fetch_reference_files(REFERENCE_REPO_ID)
+        if not reference_map:
+            raise HTTPException(status_code=500, detail="Could not fetch reference files")
+        
+        # Get file URL
+        audio_url = hf_hub_url(repo_id=SOURCE_REPO_ID, filename=filename, repo_type="dataset", subfolder=None)
+        local_wav_path = os.path.join(DOWNLOAD_FOLDER, os.path.basename(filename))
+        
+        # Download
+        if not download_with_retry(audio_url, local_wav_path):
+            raise HTTPException(status_code=500, detail="Failed to download file")
+        
+        # Find match
+        base_filename = os.path.basename(filename)
+        matched_filename = find_matching_filename(base_filename, reference_map)
+        
+        if not matched_filename:
+            os.remove(local_wav_path)
+            raise HTTPException(status_code=404, detail="No matching filename found")
+        
+        # Process
+        if process_audio_file(local_wav_path, reference_map, matched_filename):
+            processing_status["transcribed_files"] += 1
+            
+            if os.path.exists(local_wav_path):
+                os.remove(local_wav_path)
+            
+            return {
+                "status": "success",
+                "file": filename,
+                "matched": matched_filename,
+                "message": "Audio transcribed and uploaded successfully"
+            }
+        else:
+            if os.path.exists(local_wav_path):
+                os.remove(local_wav_path)
+            raise HTTPException(status_code=500, detail="Processing failed")
+            
+    except Exception as e:
+        log_message(f"❌ Manual processing error: {str(e)}", "ERROR")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.on_event("startup")
+async def startup_event():
+    """Auto-start processing when server starts"""
+    log_message("🚀 Server startup: Auto-starting processing loop", "INFO")
+    
+    # Start processing in a separate thread
+    thread = threading.Thread(target=main_processing_loop, daemon=True)
+    thread.start()
+
+def run_api(host: str = "0.0.0.0", port: int = 8000):
+    """Run the FastAPI server"""
+    log_message(f"🚀 Starting FastAPI server on {host}:{port}", "INFO")
+    uvicorn.run(app, host=host, port=port)
+
+if __name__ == "__main__":
+    # Run API server (processing will auto-start via startup event)
+    run_api()