Update app.py

app.py

@@ -4,7 +4,6 @@ from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.messages import HumanMessage, SystemMessage
 from typing import TypedDict, Annotated, List, Optional
 import json
-import time
 
 # Configuration
 class MedicalConfig:
@@ -41,7 +40,9 @@ class MedicalAgent:
 
     def analyze_patient(self, patient_data: dict) -> Optional[dict]:
         try:
-            response = self.model.invoke([
+            # Bind tools to the model
+            model_with_tools = self.model.bind_tools(list(self.tools["medical_actions"].keys()))
+            response = model_with_tools.invoke([
                 SystemMessage(content=MedicalConfig.SYSTEM_PROMPT),
                 HumanMessage(content=f"Patient Data: {json.dumps(patient_data)}")
             ])
@@ -54,7 +55,7 @@ class MedicalAgent:
         try:
             if action['name'] in self.tools['medical_actions']:
                 return self.tools['medical_actions'][action['name']](action['args'])
-            return "Unknown action"
+            return f"Unknown action: {action['name']}"
         except Exception as e:
             return f"Error processing action: {str(e)}"
 
@@ -109,12 +110,14 @@ def main():
             response = st.session_state.agent.analyze_patient(patient_data)
             
             if response:
-                if hasattr(response, 'tool_calls'):
+                st.subheader("Analysis Results")
+                if hasattr(response, 'tool_calls') and response.tool_calls:
                     for action in response.tool_calls:
                         result = st.session_state.agent.process_action(action)
                         st.success(result)
                 else:
-                    st.info(response.content)
+                    st.info("No specific actions recommended. Here's the analysis:")
+                    st.write(response.content)
 
 # Run the app
 if __name__ == "__main__":