midterm-submission

server/main.py

@@ -39,10 +39,29 @@ chat_model = ChatOpenAI(
     openai_api_key=openai_api_key
 )
 
+# Define Pydantic models
+class ChatMessage(BaseModel):
+    content: str
+    context: Optional[Dict[str, Any]] = None
+    agent_type: Optional[str] = "believer"
+
+class WorkflowResponse(BaseModel):
+    debate_history: List[Dict[str, str]]
+    supervisor_notes: List[str]
+    supervisor_chunks: List[Dict[str, List[str]]]
+    extractor_data: Dict[str, Any]
+    final_podcast: Dict[str, Any]
+
+class PodcastChatRequest(BaseModel):
+    message: str
+
+class PodcastChatResponse(BaseModel):
+    response: str
+
 # Initialize FastAPI app
 app = FastAPI()
 
-# Configure CORS for frontend development server
+# Configure CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["http://localhost:5173", "http://localhost:3000", "https://*.hf.space", "*"],
@@ -53,37 +72,49 @@ app.add_middleware(
     max_age=600,
 )
 
-# Mount static files for React frontend
-app.mount("/", StaticFiles(directory="static", html=True), name="frontend")
-
-# Configure audio storage
+# Configure storage directories
 audio_dir = os.path.join(os.path.dirname(__file__), "audio_storage")
 os.makedirs(audio_dir, exist_ok=True)
 
-# Mount the audio directory as a static file directory
-app.mount("/audio-files", StaticFiles(directory=audio_dir), name="audio")
-
-# Configure context storage
 context_dir = os.path.join(os.path.dirname(__file__), "context_storage")
 os.makedirs(context_dir, exist_ok=True)
 
-class ChatMessage(BaseModel):
-    content: str
-    context: Optional[Dict[str, Any]] = None
-    agent_type: Optional[str] = "believer"
-
-class WorkflowResponse(BaseModel):
-    debate_history: List[Dict[str, str]]
-    supervisor_notes: List[str]
-    supervisor_chunks: List[Dict[str, List[str]]]
-    extractor_data: Dict[str, Any]
-    final_podcast: Dict[str, Any]
-
-class PodcastChatRequest(BaseModel):
-    message: str
+# API Routes
+@app.post("/chat")
+async def chat(message: ChatMessage):
+    """Process a chat message."""
+    try:
+        # Get API key
+        tavily_api_key = os.getenv("TAVILY_API_KEY")
+        if not tavily_api_key:
+            logger.error("Tavily API key not found")
+            raise HTTPException(status_code=500, detail="Tavily API key not configured")
 
-class PodcastChatResponse(BaseModel):
-    response: str
+        # Initialize the workflow
+        try:
+            workflow = create_workflow(tavily_api_key)
+            logger.info("Workflow created successfully")
+        except Exception as e:
+            logger.error(f"Error creating workflow: {str(e)}")
+            raise HTTPException(status_code=500, detail=f"Error creating workflow: {str(e)}")
+        
+        # Run the workflow with context
+        try:
+            result = await run_workflow(
+                workflow,
+                message.content,
+                agent_type=message.agent_type,
+                context=message.context
+            )
+            logger.info("Workflow completed successfully")
+            return result
+        except Exception as e:
+            logger.error(f"Error running workflow: {str(e)}")
+            raise HTTPException(status_code=500, detail=f"Error running workflow: {str(e)}")
+            
+    except Exception as e:
+        logger.error(f"Error in chat endpoint: {str(e)}", exc_info=True)
+        raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/audio-list")
 async def list_audio_files():
@@ -103,55 +134,7 @@ async def list_audio_files():
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.delete("/audio/{filename}")
-async def delete_audio_file(filename: str):
-    """Delete an audio file and its corresponding transcript."""
-    try:
-        # Delete audio file
-        file_path = os.path.join(audio_dir, filename)
-        if not os.path.exists(file_path):
-            raise HTTPException(status_code=404, detail="File not found")
-        
-        # Get all audio files to determine the podcast ID
-        audio_files = [f for f in os.listdir(audio_dir) if f.endswith(('.mp3', '.wav'))]
-        try:
-            # Find the index (0-based) of the file being deleted
-            podcast_id = audio_files.index(filename) + 1  # Convert to 1-based ID
-            logger.info(f"Deleting podcast with ID: {podcast_id}")
-            
-            # Path to transcripts file
-            transcripts_file = os.path.join(os.path.dirname(__file__), "transcripts", "podcasts.json")
-            
-            # Update transcripts if file exists
-            if os.path.exists(transcripts_file):
-                with open(transcripts_file, 'r') as f:
-                    transcripts = json.load(f)
-                
-                # Remove the transcript at the corresponding index
-                if len(transcripts) >= podcast_id:
-                    transcripts.pop(podcast_id - 1)  # Convert back to 0-based index
-                    
-                    # Save updated transcripts
-                    with open(transcripts_file, 'w') as f:
-                        json.dump(transcripts, f, indent=2)
-                    logger.info(f"Removed transcript for podcast ID {podcast_id}")
-            
-            # Delete the audio file
-            os.remove(file_path)
-            logger.info(f"Deleted audio file: {filename}")
-            
-            return {"message": "File and transcript deleted successfully"}
-            
-        except ValueError:
-            logger.error(f"Could not determine podcast ID for file: {filename}")
-            # Still delete the audio file even if transcript removal fails
-            os.remove(file_path)
-            return {"message": "Audio file deleted, but transcript could not be removed"}
-            
-    except Exception as e:
-        logger.error(f"Error in delete_audio_file: {str(e)}")
-        raise HTTPException(status_code=500, detail=str(e))
-
+# Add other API routes here...
 @app.get("/audio/{filename}")
 async def get_audio_file(filename: str):
     """Get an audio file by filename."""
@@ -288,42 +271,6 @@ async def get_podcast_context(podcast_id: str):
         logger.error(f"Error in get_podcast_context: {str(e)}", exc_info=True)
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.post("/chat")
-async def chat(message: ChatMessage):
-    """Process a chat message."""
-    try:
-        # Get API key
-        tavily_api_key = os.getenv("TAVILY_API_KEY")
-        if not tavily_api_key:
-            logger.error("Tavily API key not found")
-            raise HTTPException(status_code=500, detail="Tavily API key not configured")
-
-        # Initialize the workflow
-        try:
-            workflow = create_workflow(tavily_api_key)
-            logger.info("Workflow created successfully")
-        except Exception as e:
-            logger.error(f"Error creating workflow: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"Error creating workflow: {str(e)}")
-        
-        # Run the workflow with context
-        try:
-            result = await run_workflow(
-                workflow,
-                message.content,
-                agent_type=message.agent_type,
-                context=message.context
-            )
-            logger.info("Workflow completed successfully")
-            return result
-        except Exception as e:
-            logger.error(f"Error running workflow: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"Error running workflow: {str(e)}")
-            
-    except Exception as e:
-        logger.error(f"Error in chat endpoint: {str(e)}", exc_info=True)
-        raise HTTPException(status_code=500, detail=str(e))
-
 @app.post("/podcast-chat/{podcast_id}", response_model=PodcastChatResponse)
 async def podcast_chat(podcast_id: str, request: PodcastChatRequest):
     """Handle chat messages for a specific podcast."""
@@ -429,6 +376,59 @@ async def podcast_chat(podcast_id: str, request: PodcastChatRequest):
         logger.error(f"Error in podcast chat: {str(e)}", exc_info=True)
         raise HTTPException(status_code=500, detail=str(e))
 
+@app.delete("/audio/{filename}")
+async def delete_audio_file(filename: str):
+    """Delete an audio file and its corresponding transcript."""
+    try:
+        # Delete audio file
+        file_path = os.path.join(audio_dir, filename)
+        if not os.path.exists(file_path):
+            raise HTTPException(status_code=404, detail="File not found")
+        
+        # Get all audio files to determine the podcast ID
+        audio_files = [f for f in os.listdir(audio_dir) if f.endswith(('.mp3', '.wav'))]
+        try:
+            # Find the index (0-based) of the file being deleted
+            podcast_id = audio_files.index(filename) + 1  # Convert to 1-based ID
+            logger.info(f"Deleting podcast with ID: {podcast_id}")
+            
+            # Path to transcripts file
+            transcripts_file = os.path.join(os.path.dirname(__file__), "transcripts", "podcasts.json")
+            
+            # Update transcripts if file exists
+            if os.path.exists(transcripts_file):
+                with open(transcripts_file, 'r') as f:
+                    transcripts = json.load(f)
+                
+                # Remove the transcript at the corresponding index
+                if len(transcripts) >= podcast_id:
+                    transcripts.pop(podcast_id - 1)  # Convert back to 0-based index
+                    
+                    # Save updated transcripts
+                    with open(transcripts_file, 'w') as f:
+                        json.dump(transcripts, f, indent=2)
+                    logger.info(f"Removed transcript for podcast ID {podcast_id}")
+            
+            # Delete the audio file
+            os.remove(file_path)
+            logger.info(f"Deleted audio file: {filename}")
+            
+            return {"message": "File and transcript deleted successfully"}
+            
+        except ValueError:
+            logger.error(f"Could not determine podcast ID for file: {filename}")
+            # Still delete the audio file even if transcript removal fails
+            os.remove(file_path)
+            return {"message": "Audio file deleted, but transcript could not be removed"}
+            
+    except Exception as e:
+        logger.error(f"Error in delete_audio_file: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+# Static file mounts (must be last)
+app.mount("/audio-files", StaticFiles(directory=audio_dir), name="audio")
+app.mount("/", StaticFiles(directory="static", html=True), name="frontend")
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)