Update app.py

app.py

@@ -51,7 +51,6 @@ def ingest_data():
     try:
         driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USERNAME, NEO4J_PASSWORD))
         
-        # PrimeKG Subset Data
         prime_kg_data = [
             ("Ischemic Stroke", "associated_with", "Hypertension"),
             ("Ischemic Stroke", "associated_with", "Atrial Fibrillation"),
@@ -79,7 +78,7 @@ def ingest_data():
         """
         
         with driver.session() as session:
-            session.run("MATCH (n) DETACH DELETE n") # Reset DB for demo
+            session.run("MATCH (n) DETACH DELETE n")
             session.run(query, data=prime_kg_data)
         
         # Unstructured Guidelines Data
@@ -110,7 +109,7 @@ def ingest_data():
     except Exception as e:
         print(f"❌ KG Initialization Failed: {e}")
 
-# Run ingestion immediately on startup
+# Run ingestion
 ingest_data()
 
 # ==========================================
@@ -128,13 +127,11 @@ def detect_stroke_type(image_path: str) -> str:
 
     try:
         client = Client(HF_SPACE_URL)
-        # Call the remote API
         result = client.predict(
             files=[handle_file(image_path)],
             api_name="/predict"
         )
         
-        # Robust parsing for Gradio returns
         ai_output = {}
         if isinstance(result, (list, tuple)) and len(result) > 1:
             json_data = result[1]
@@ -242,7 +239,6 @@ def synthesizer_node(state: AgentState):
     response = llm.invoke(prompt)
     return {"final_report": response.content}
 
-# Build Graph
 workflow = StateGraph(AgentState)
 workflow.add_node("detector", detector_node)
 workflow.add_node("advisor", advisor_node)
@@ -271,7 +267,7 @@ def predict_stroke_report(uploaded_file, patient_context_input):
     try:
         # Run Graph
         for event in app.stream(inputs, stream_mode="values"):
-            pass # Exhaust stream
+            pass 
         final_state = event
         final_report_content = final_state.get("final_report", "Error generating report.")
     except Exception as e:
@@ -279,12 +275,10 @@ def predict_stroke_report(uploaded_file, patient_context_input):
 
     return final_report_content
 
-# Define Theme
 custom_theme = gr.themes.Soft(primary_hue="emerald", secondary_hue="neutral")
 
-# Launch App
-# FIX: Removed title="" from Blocks() to ensure compatibility.
-with gr.Blocks(theme=custom_theme) as demo:
+# Since we updated the SDK version to 5.0.0, the theme argument is now valid.
+with gr.Blocks(theme=custom_theme, title="Stroke Agent AI") as demo:
     gr.Markdown("# 🧠 AI Stroke Diagnosis & Guideline Agent")
     gr.Markdown("Integrates Vision AI (DenseNet) + Knowledge Graph (Neo4j) + LLM (Azure OpenAI)")