Update app.py

app.py

@@ -37,14 +37,12 @@ def process_chat(message, history, session_id):
         thread_id = client.beta.threads.create().id
         session_threads[session_id] = thread_id
 
-    # Store user message
     client.beta.threads.messages.create(
         thread_id=thread_id,
         role="user",
         content=message
     )
 
-    # Run assistant
     run = client.beta.threads.runs.create(
         thread_id=thread_id,
         assistant_id=ASSISTANT_ID
@@ -59,7 +57,6 @@ def process_chat(message, history, session_id):
             break
         time.sleep(1)
 
-    # Retrieve assistant message
     messages = client.beta.threads.messages.list(thread_id=thread_id)
     for msg in reversed(messages.data):
         if msg.role == "assistant":
@@ -68,7 +65,7 @@ def process_chat(message, history, session_id):
     else:
         assistant_response = "⚠️ Assistant did not respond."
 
-    # Detect image if present
+    # Detect image from GitHub
     image_url = None
     match = re.search(r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png', assistant_response)
     if match:
@@ -115,10 +112,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             ),
             additional_inputs=[session_id],
             examples=["What does clause 3.2 mean?", "Summarize the timeline from the image."],
-            title="💬 Document Assistant",
-            retry_btn="🔁 Retry",
-            undo_btn="↩️ Undo",
-            clear_btn="🗑️ Clear",
+            title="💬 Document Assistant"
         )
 
     # Link image preview if extracted