Update app.py

app.py

@@ -264,23 +264,23 @@ theme = gr.themes.Ocean(
 demo_css = """
 .chatbot {
     font-family: 'Inter', 'Segoe UI', sans-serif !important;
-    font-size: 12px !important;
+    font-size: 13px !important;
 }
 
 .chatbot .message code,
 .chatbot .message pre {
-    font-size: 12px !important;
+    font-size: 13px !important;
     font-family: 'Inter', 'Segoe UI', sans-serif !important;
 }
 
 .component {
     font-family: 'Inter', 'Segoe UI', sans-serif !important;
-    font-size: 12px !important;
+    font-size: 13px !important;
 }
 
 .gradio-container .examples {
     font-family: 'Inter', 'Segoe UI', sans-serif !important;
-    font-size: 12px !important;
+    font-size: 13px !important;
 }
 """
 
@@ -315,6 +315,7 @@ with gr.Blocks(
             #ask_btn = gr.Button("Submit 📤", variant="primary")
             with gr.Row():
                 ask_btn = gr.Button("Submit 📤", variant="primary", elem_classes="component")
+                regen_btn = gr.Button("🔄 Regenerate", elem_classes="component")
                 clear_btn = gr.Button("🧹 Clear Chat", elem_classes="component")
             # A section for example questions.
             gr.Markdown("### 💡 Example Questions", elem_classes="component")
@@ -402,7 +403,22 @@ with gr.Blocks(
             gr.update(value=None, visible=False), # Keep image preview hidden for now.
             "" # Keep textbox clear.
         )
-        
+
+    def regen_wrapper(chat_history):
+        if not chat_history or len(chat_history) < 2:
+            return chat_history, None, None, ""
+    
+        last_user_msg = chat_history[-2]["content"]
+        new_history = chat_history[:-1]
+    
+        return chat_ui(last_user_msg, new_history, "Hybrid")
+
+    regen_btn.click(
+        regen_wrapper,
+        inputs=chatbot,
+        outputs=[chatbot, link_state, output_image, question]
+    )
+    
     # This is a helper function to avoid repeating the event handler chain.
     def submit_chain():
         # It specifies that `chat_ui` is the function to run.