Rollback to basic UI

app.py

@@ -2,121 +2,59 @@ import gradio as gr
 import uuid
 from eldersafe_pipeline import run_eldersafe
 
+# Generate one session ID per app run
+SESSION_ID = str(uuid.uuid4())
 
-# -------------------------
-# Session ID Helpers
-# -------------------------
-def get_or_create_session_id(cookie):
-    if cookie and "session_id" in cookie:
-        return cookie["session_id"]
-    return str(uuid.uuid4())
-
-
-def reset_session():
-    return (
-        str(uuid.uuid4()),
-        [],
-        "",
-    )
-
-
-# -------------------------
-# Main chat handler
-# -------------------------
-async def eldersafe_chat(history, message, session_id):
+# Gradio supports async functions, so we can await the pipeline directly.
+async def eldersafe_interface(message: str):
     if not message.strip():
-        return history, session_id
+        return "Please paste a WhatsApp / social media message to analyze."
 
-    history.append(("🧑‍💬 You", message))
-
-    result = await run_eldersafe(message, session_id)
+    # Use auto-generated session ID for all calls
+    result = await run_eldersafe(message, SESSION_ID)
 
     clean_claim = result.get("clean_claim", "")
     final_report = result.get("final_report", "")
     memory_context = result.get("memory_context", "")
 
-    reply = f"""
-### 🔍 **Claim Analyzed**
-> *{clean_claim}*
+    md = f"### 🔍 Viral Message Analyzed\n> *{clean_claim}*\n\n"
+    md += f"{final_report}\n"
 
-{final_report}
-"""
+    if memory_context and "No previous checks" not in str(memory_context):
+        md += "\n---\n**📜 Your Recent Checks:**\n"
+        md += str(memory_context)
 
-    if memory_context and "No previous checks" not in memory_context:
-        reply += f"""
----
-### 📜 **Your Recent Checks**
-{memory_context}
-"""
+    return md
 
-    history.append(("🤖 ElderSafe", reply))
 
-    return history, session_id
+title = "🧓 ElderSafe – Fake News Evidence Mapping Agent"
+description = """
+Paste any WhatsApp or social media forward, and ElderSafe will:
+- Extract the main claim
+- Search for evidence
+- Show a clear verdict
+- Explain it in simple terms for elderly users
+"""
 
-# -------------------------
-# BUILD UI (VERSION SAFE)
-# -------------------------
 with gr.Blocks() as demo:
+    gr.Markdown(f"# {title}")
+    gr.Markdown(description)
 
-    cookie_state = gr.State({})
-    session_box = gr.State("")
-
-    gr.HTML("""
-    <div style="text-align:center; padding:18px; font-family: sans-serif;">
-        <h1 style="color:#008080; margin-bottom: 6px;">🛡️ ElderSafe</h1>
-        <h3 style="color:#444; margin-top:0;">Fake News Evidence Mapping Assistant</h3>
-        <p style="color:#666; font-size:15px;">
-            A warm, simple WhatsApp-like chat to help verify WhatsApp forwards.
-        </p>
-    </div>
-    """)
-
-    with gr.Row():
-        new_session_btn = gr.Button("🆕 New Analysis")
-
-    chatbot = gr.Chatbot(label="ElderSafe Assistant")
-
-    # -------------------------
-    # FLOATING SEND BUTTON (Option C)
-    # -------------------------
-    with gr.Row():
-        message_input = gr.Textbox(
-            placeholder="Paste any WhatsApp message here…",
-            label="Your Message",
-            lines=3,
-            scale=8,    # makes textbox wider
-        )
-
-        send_btn = gr.Button(
-            "Send",
-            variant="primary",
-            scale=1     # keeps button small and floating-like
-        )
-
-    # Submit via ENTER
-    message_input.submit(
-        eldersafe_chat,
-        inputs=[chatbot, message_input, session_box],
-        outputs=[chatbot, session_box],
+    msg = gr.Textbox(
+        label="Paste WhatsApp / Social Media Message",
+        lines=6,
+        placeholder="e.g. NASA has selected an Indian Ayurvedic doctor for space medicine research.",
     )
 
-    # Submit via SEND button
-    send_btn.click(
-        eldersafe_chat,
-        inputs=[chatbot, message_input, session_box],
-        outputs=[chatbot, session_box],
-    )
+    output = gr.Markdown(label="Verification Report")
 
-    # Reset session
-    new_session_btn.click(
-        reset_session,
-        inputs=None,
-        outputs=[session_box, chatbot, chatbot],
-    )
+    analyze_btn = gr.Button("🔎 Analyze Message")
 
-    # Initialize session ID from cookie
-    demo.load(
-        get_or_create_session_id,
-        inputs=cookie_state,
-        outputs=session_box,
+    analyze_btn.click(
+        fn=eldersafe_interface,
+        inputs=msg,
+        outputs=output,
     )
+
+if __name__ == "__main__":
+    demo.launch()