Revert portg

app.py

@@ -9,7 +9,7 @@ import urllib.request
 import torchaudio
 from scipy.spatial.distance import cosine
 from RealtimeSTT import AudioToTextRecorder
-from fastapi import FastAPI
+from fastapi import FastAPI, APIRouter
 from fastrtc import Stream, AsyncStreamHandler, ReplyOnPause, get_cloudflare_turn_credentials_async, get_cloudflare_turn_credentials
 import json
 import io
@@ -788,6 +788,50 @@ def create_interface():
     return interface
 
 
+# Create API router for endpoints
+router = APIRouter()
+
+# Health check endpoint
+@router.get("/health")
+async def health_check():
+    """Health check endpoint"""
+    return {
+        "status": "healthy",
+        "timestamp": time.time(),
+        "system_initialized": diarization_system.encoder is not None,
+        "recording_active": diarization_system.is_running
+    }
+
+
+# API endpoint to get conversation
+@router.get("/api/conversation")
+async def get_conversation_api():
+    """API endpoint to get current conversation"""
+    return {
+        "conversation": diarization_system.get_formatted_conversation(),
+        "status": diarization_system.get_status_info(),
+        "is_recording": diarization_system.is_running
+    }
+
+
+# API endpoint to control recording
+@router.post("/api/control/{action}")
+async def control_recording(action: str):
+    """API endpoint to control recording (start/stop/clear)"""
+    if action == "start":
+        result = diarization_system.start_recording()
+    elif action == "stop":
+        result = diarization_system.stop_recording()
+    elif action == "clear":
+        result = diarization_system.clear_conversation()
+    elif action == "initialize":
+        result = initialize_system()
+    else:
+        return {"error": "Invalid action. Use: start, stop, clear, or initialize"}
+    
+    return {"result": result, "is_recording": diarization_system.is_running}
+
+
 # Main application setup
 def create_app():
     """Create and configure the FastAPI app with Gradio and FastRTC"""
@@ -798,6 +842,9 @@ def create_app():
         version="1.0.0"
     )
     
+    # Include API routes
+    app.include_router(router)
+    
     # Create Gradio interface
     gradio_interface = create_interface()
     
@@ -851,47 +898,6 @@ def create_app():
     return app
 
 
-# Health check endpoint
-@app.get("/health")
-async def health_check():
-    """Health check endpoint"""
-    return {
-        "status": "healthy",
-        "timestamp": time.time(),
-        "system_initialized": diarization_system.encoder is not None,
-        "recording_active": diarization_system.is_running
-    }
-
-
-# API endpoint to get conversation
-@app.get("/api/conversation")
-async def get_conversation_api():
-    """API endpoint to get current conversation"""
-    return {
-        "conversation": diarization_system.get_formatted_conversation(),
-        "status": diarization_system.get_status_info(),
-        "is_recording": diarization_system.is_running
-    }
-
-
-# API endpoint to control recording
-@app.post("/api/control/{action}")
-async def control_recording(action: str):
-    """API endpoint to control recording (start/stop/clear)"""
-    if action == "start":
-        result = diarization_system.start_recording()
-    elif action == "stop":
-        result = diarization_system.stop_recording()
-    elif action == "clear":
-        result = diarization_system.clear_conversation()
-    elif action == "initialize":
-        result = initialize_system()
-    else:
-        return {"error": "Invalid action. Use: start, stop, clear, or initialize"}
-    
-    return {"result": result, "is_recording": diarization_system.is_running}
-
-
 # Main entry point
 if __name__ == "__main__":
     # Create the app