Update app.py

app.py

@@ -38,7 +38,7 @@ if missing_keys:
 
 # --- Configuration & Constants ---
 class ClinicalAppSettings:
-    APP_TITLE = "SynapseAI "
+    APP_TITLE = "SynapseAI (UMLS/FDA Integrated)"
     PAGE_LAYOUT = "wide"
     MODEL_NAME = "llama3-70b-8192"
     TEMPERATURE = 0.1
@@ -437,10 +437,14 @@ def tool_node(state: AgentState):
     for med_name, prescribe_call in prescriptions_requested.items():
         if med_name not in interaction_checks_requested:
             st.error(f"**Safety Violation:** AI tried to prescribe '{med_name}' without check.")
-            error_msg = ToolMessage(content=json.dumps({
-                "status": "error",
-                "message": f"Interaction check needed for '{med_name}'."
-            }), tool_call_id=prescribe_call['id'], name=prescribe_call['name'])
+            error_msg = ToolMessage(
+                content=json.dumps({
+                    "status": "error",
+                    "message": f"Interaction check needed for '{med_name}'."
+                }),
+                tool_call_id=prescribe_call['id'],
+                name=prescribe_call['name']
+            )
             tool_messages.append(error_msg)
             blocked_ids.add(prescribe_call['id'])
     valid_tool_calls_for_execution = [call for call in tool_calls if call['id'] not in blocked_ids]
@@ -467,17 +471,27 @@ def tool_node(state: AgentState):
                 print(f"ERROR executing tool '{tool_name}': {error_type} - {error_str}")
                 traceback.print_exc()
                 st.error(f"Error: {error_type}")
-                error_content = json.dumps({"status": "error", "message": f"Failed: {error_type} - {error_str}"})
+                error_content = json.dumps({
+                    "status": "error",
+                    "message": f"Failed: {error_type} - {error_str}"
+                })
                 tool_messages.append(ToolMessage(content=error_content, tool_call_id=tool_call_id, name=tool_name))
             else:
                 print(f"Tool '{tool_name}' executed.")
-                content_str = str(resp)
+                # If response is a dict, convert it to a JSON string.
+                if isinstance(resp, dict):
+                    content_str = json.dumps(resp)
+                else:
+                    content_str = str(resp)
                 tool_messages.append(ToolMessage(content=content_str, tool_call_id=tool_call_id, name=tool_name))
     except Exception as e:
         print(f"CRITICAL TOOL NODE ERROR: {e}")
         traceback.print_exc()
         st.error(f"Critical error: {e}")
-        error_content = json.dumps({"status": "error", "message": f"Internal error: {e}"})
+        error_content = json.dumps({
+            "status": "error",
+            "message": f"Internal error: {e}"
+        })
         processed_ids = {msg.tool_call_id for msg in tool_messages}
         [tool_messages.append(ToolMessage(content=error_content, tool_call_id=call['id'], name=call['name']))
          for call in valid_tool_calls_for_execution if call['id'] not in processed_ids]
@@ -598,7 +612,7 @@ def main():
 
     # --- Main Chat Interface Area ---
     st.header("💬 Clinical Consultation")
-    # Display loop - SyntaxError Fixed
+    # Display loop
     for msg in st.session_state.messages:
         if isinstance(msg, HumanMessage):
             with st.chat_message("user"):
@@ -693,7 +707,7 @@ def main():
                     message = tool_data.get("message", msg.content)
                     details = tool_data.get("details")
                     warnings = tool_data.get("warnings")
-                    if status == "success" or status == "clear" or status == "flagged":
+                    if status in ["success", "clear", "flagged"]:
                         st.success(f"{message}", icon="✅" if status != "flagged" else "🚨")
                     elif status == "warning":
                         st.warning(f"{message}", icon="⚠️")