Update app.py

app.py

@@ -65,7 +65,7 @@ def process_chat(message, history, session_id):
     else:
         assistant_response = "⚠️ Assistant did not respond."
 
-    # Detect image from GitHub
+    # Optional: detect GitHub image URL
     image_url = None
     match = re.search(r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png', assistant_response)
     if match:
@@ -99,11 +99,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     session_id = gr.State(value=reset_session())
     client_id = gr.State()
 
-    # ---------- Section 1: Document Image Display ----------
-    with gr.Row():
-        image_display = gr.Image(label="📄 Document Page (auto-extracted if available)", interactive=False, visible=False)
-
-    # ---------- Section 2: Chat Interface ----------
+    # ---------- Section 1: Chat Interface ----------
     with gr.Row():
         chatbot = gr.ChatInterface(
             fn=lambda message, history, session_id: (
@@ -118,14 +114,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             title="💬 Document Assistant"
         )
 
-    # Link image preview if extracted
-    def update_image_display(message, history, session_id):
-        _, image_url = process_chat(message, history, session_id)
-        return gr.update(value=image_url, visible=bool(image_url))
-
-    chatbot.chatbot.change(fn=update_image_display, inputs=[chatbot.input, chatbot.chatbot, session_id], outputs=[image_display])
-
-    # ---------- Section 3: Voice Transcription ----------
+    # ---------- Section 2: Voice Transcription ----------
     gr.Markdown("## 🎙️ Realtime Voice Transcription")
 
     with gr.Row():