midterm-submission

server/main.py

@@ -290,11 +290,8 @@ async def get_podcast_context(podcast_id: str):
 
 @app.post("/chat")
 async def chat(message: ChatMessage):
-    """Process a chat message with context-awareness."""
+    """Process a chat message."""
     try:
-        # Log incoming message
-        logger.info(f"Received chat message: {message}")
-        
         # Get API key
         tavily_api_key = os.getenv("TAVILY_API_KEY")
         if not tavily_api_key:
@@ -318,11 +315,11 @@ async def chat(message: ChatMessage):
                 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)}")
-        
-        return WorkflowResponse(**result)
+            
     except Exception as e:
         logger.error(f"Error in chat endpoint: {str(e)}", exc_info=True)
         raise HTTPException(status_code=500, detail=str(e))