Update app.py

app.py

@@ -93,6 +93,28 @@ except Exception:
     def get_category_for_mood(m): return "neutral"
     def get_emoji(cat, intensity=0.5): return "🤖"
 
+# Import custom modules
+try:
+    from voicecloner import synthesize_speech, is_available as tts_is_available, cache_speaker_sample
+    VOICECLONER_AVAILABLE = True
+except Exception:
+    VOICECLONER_AVAILABLE = False
+    logger.warning("voicecloner module not available")
+
+try:
+    from coder import Coder
+    CODER_AVAILABLE = True
+except Exception:
+    CODER_AVAILABLE = False
+    logger.warning("coder module not available")
+
+try:
+    from videogenerator import VideoGenerator
+    VIDEOGEN_AVAILABLE = True
+except Exception:
+    VIDEOGEN_AVAILABLE = False
+    logger.warning("videogenerator module not available")
+
 # Optional langdetect
 try:
     from langdetect import detect as detect_lang
@@ -162,6 +184,10 @@ engine_knowledge = create_engine(
 
 app = FastAPI(title="Justice Brain — Backend")
 
+# Initialize custom modules
+coder_instance = Coder() if CODER_AVAILABLE else None
+video_generator = VideoGenerator() if VIDEOGEN_AVAILABLE else None
+
 # -------------------------
 # Database schema creation
 # -------------------------
@@ -1305,7 +1331,10 @@ async def model_status():
         "language_module": LANGUAGE_MODULE_AVAILABLE,
         "tts_available": TTS_AVAILABLE,
         "multipart_available": HAVE_MULTIPART,
-        "pillow_available": PIL_AVAILABLE
+        "pillow_available": PIL_AVAILABLE,
+        "voicecloner_available": VOICECLONER_AVAILABLE,
+        "coder_available": CODER_AVAILABLE,
+        "videogen_available": VIDEOGEN_AVAILABLE
     }
 
 @app.get("/health")
@@ -1556,6 +1585,199 @@ async def clear_database(data: dict = Body(...), x_admin_key: str = Header(None,
         logger.exception("clear failed")
         return JSONResponse(status_code=500, content={"error": "failed to clear database", "details": str(e)})
 
+# -------------------------
+# Coder endpoints
+# -------------------------
+@app.post("/coder/run")
+async def coder_run_code(data: dict = Body(...)):
+    """Execute code in sandbox"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    code = data.get("code", "")
+    lang = data.get("language", "python")
+    timeout = int(data.get("timeout", 15))
+    
+    if not code:
+        raise HTTPException(status_code=400, detail="Code is required")
+    
+    try:
+        result = coder_instance.run_code(code, lang, timeout)
+        return result
+    except Exception as e:
+        logger.exception("Coder run failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.post("/coder/debug")
+async def coder_debug_code(data: dict = Body(...)):
+    """Debug code in sandbox"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    code = data.get("code", "")
+    lang = data.get("language", "python")
+    
+    if not code:
+        raise HTTPException(status_code=400, detail="Code is required")
+    
+    try:
+        result = coder_instance.debug_code(code, lang)
+        return result
+    except Exception as e:
+        logger.exception("Coder debug failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.post("/coder/preview/start")
+async def coder_start_preview(data: dict = Body(...)):
+    """Start preview server"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    lang = data.get("language", "html")
+    port = int(data.get("port", 8000))
+    
+    try:
+        result = coder_instance.start_preview(lang, port)
+        return result
+    except Exception as e:
+        logger.exception("Preview start failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.post("/coder/preview/stop")
+async def coder_stop_preview():
+    """Stop preview server"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    try:
+        result = coder_instance.stop_preview()
+        return result
+    except Exception as e:
+        logger.exception("Preview stop failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.get("/coder/preview/info")
+async def coder_preview_info():
+    """Get preview server info"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    try:
+        result = coder_instance.get_preview_info()
+        return result
+    except Exception as e:
+        logger.exception("Preview info failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.post("/coder/file/write")
+async def coder_write_file(data: dict = Body(...)):
+    """Write file to sandbox"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    filename = data.get("filename", "")
+    content = data.get("content", "")
+    
+    if not filename:
+        raise HTTPException(status_code=400, detail="Filename is required")
+    
+    try:
+        result = coder_instance.write_file(filename, content)
+        return result
+    except Exception as e:
+        logger.exception("File write failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.post("/coder/file/read")
+async def coder_read_file(data: dict = Body(...)):
+    """Read file from sandbox"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    filename = data.get("filename", "")
+    
+    if not filename:
+        raise HTTPException(status_code=400, detail="Filename is required")
+    
+    try:
+        result = coder_instance.read_file(filename)
+        return result
+    except Exception as e:
+        logger.exception("File read failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.get("/coder/files")
+async def coder_list_files():
+    """List files in sandbox"""
+    if not CODER_AVAILABLE or coder_instance is None:
+        raise HTTPException(status_code=503, detail="Coder module not available")
+    
+    try:
+        result = coder_instance.list_files()
+        return result
+    except Exception as e:
+        logger.exception("File list failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+# -------------------------
+# Video Generator endpoints
+# -------------------------
+@app.post("/video/generate")
+async def video_generate(background_tasks: BackgroundTasks, data: dict = Body(...)):
+    """Generate video from prompt"""
+    if not VIDEOGEN_AVAILABLE or video_generator is None:
+        raise HTTPException(status_code=503, detail="Video generator not available")
+    
+    prompt = data.get("prompt", "")
+    num_frames = int(data.get("num_frames", 16))
+    fps = int(data.get("fps", 8))
+    enhance = bool(data.get("enhance", False))
+    
+    if not prompt:
+        raise HTTPException(status_code=400, detail="Prompt is required")
+    
+    try:
+        loop = asyncio.get_running_loop()
+        result = await loop.run_in_executor(
+            None,
+            lambda: video_generator.generate(
+                prompt=prompt,
+                num_frames=num_frames,
+                fps=fps,
+                enhance=enhance
+            )
+        )
+        return result
+    except Exception as e:
+        logger.exception("Video generation failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.get("/video/history")
+async def video_history(limit: int = Query(20)):
+    """Get video generation history"""
+    if not VIDEOGEN_AVAILABLE or video_generator is None:
+        raise HTTPException(status_code=503, detail="Video generator not available")
+    
+    try:
+        history = video_generator.get_history(limit)
+        return {"history": history}
+    except Exception as e:
+        logger.exception("Video history failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.get("/video/status")
+async def video_status():
+    """Get video generator status"""
+    if not VIDEOGEN_AVAILABLE or video_generator is None:
+        raise HTTPException(status_code=503, detail="Video generator not available")
+    
+    try:
+        status = video_generator.get_status()
+        return status
+    except Exception as e:
+        logger.exception("Video status failed")
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
 @app.get("/", response_class=HTMLResponse)
 async def frontend_dashboard():
     try: