fix

main.py

@@ -535,7 +535,7 @@ def main():
 
         files = gr.Files(label="Upload your files", file_types=definitions.ALLOWED_TYPES)
         question = gr.Textbox(label="Ask a question", lines=2, placeholder="Type your question here...")
-        chat = gr.Chatbot(label="Answers", elem_id="chat-history")
+        chat = gr.Chatbot(label="Answers", elem_id="chat-history", type="messages")
         submit_btn = gr.Button("Get Answer", variant="primary")
         processing_message = gr.HTML("", elem_id="processing-message", visible=False)
         doc_context_display = gr.Markdown("*Submit a question to see which document sections were referenced*", elem_classes="doc-context", visible=False)
@@ -569,8 +569,10 @@ def main():
             )
             try:
                 if not question_text.strip():
-                    chat_history.append({"role": "user", "content": question_text})
-                    chat_history.append({"role": "assistant", "content": "Please enter a question."})
+                    chat_history.extend([
+                        {"role": "user", "content": question_text},
+                        {"role": "assistant", "content": "Please enter a question."}
+                    ])
                     yield (
                         chat_history,
                         gr.update(visible=False),
@@ -583,8 +585,10 @@ def main():
                     )
                     return
                 if not uploaded_files:
-                    chat_history.append({"role": "user", "content": question_text})
-                    chat_history.append({"role": "assistant", "content": "Please upload at least one document."})
+                    chat_history.extend([
+                        {"role": "user", "content": question_text},
+                        {"role": "assistant", "content": "Please upload at least one document."}
+                    ])
                     yield (
                         chat_history,
                         gr.update(visible=False),
@@ -639,10 +643,7 @@ def main():
                             gr.update(interactive=False),
                             gr.update(interactive=False),
                             gr.update(interactive=False),
-                            gr.update(value='''<div style="background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">
-  <span id="processing-msg"></span>
-  <span id="processing-timer" style="opacity:0.8; margin-left:8px;"></span>
-</div>''', visible=True)
+                            gr.update(value='''<div style=\"background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);\">\n  <span id=\"processing-msg\"></span>\n  <span id=\"processing-timer\" style=\"opacity:0.8; margin-left:8px;\"></span>\n</div>''', visible=True)
                         )
                 # After all chunks, show 100%                    
                 yield (
@@ -653,10 +654,7 @@ def main():
                     gr.update(interactive=False),
                     gr.update(interactive=False),
                     gr.update(interactive=False),
-                    gr.update(value='''<div style="background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">
-  <span id="processing-msg"></span>
-  <span id="processing-timer" style="opacity:0.8; margin-left:8px;"></span>
-</div>''', visible=True)
+                    gr.update(value='''<div style=\"background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">\n  <span id=\"processing-msg\"></span>\n  <span id=\"processing-timer\" style=\"opacity:0.8; margin-left:8px;\"></span>\n</div>''', visible=True)
                 )
                 # Stage 3: Building Retriever
                 yield (
@@ -684,10 +682,7 @@ def main():
                     gr.update(interactive=False),
                     gr.update(interactive=False),
                     gr.update(interactive=False),
-                    gr.update(value='''<div style="background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">
-  <span id="processing-msg"></span>
-  <span id="processing-timer" style="opacity:0.8; margin-left:8px;"></span>
-</div>''', visible=True)
+                    gr.update(value='''<div style=\"background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);\">\n  <span id=\"processing-msg\"></span>\n  <span id=\"processing-timer\" style=\"opacity:0.8; margin-left:8px;\"></span>\n</div>''', visible=True)
                 )
                 result = orchestrator.run_workflow(question=question_text, retriever=retriever)
                 answer = result["draft_answer"]
@@ -700,16 +695,14 @@ def main():
                     gr.update(interactive=False),
                     gr.update(interactive=False),
                     gr.update(interactive=False),
-                    gr.update(value='''<div style="background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">
-  <span id="processing-msg"></span>
-  <span id="processing-timer" style="opacity:0.8; margin-left:8px;"></span>
-</div>''', visible=True)
+                    gr.update(value='''<div style=\"background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);\">\n  <span id=\"processing-msg\"></span>\n  <span id=\"processing-timer\" style=\"opacity:0.8; margin-left:8px;\"></span>\n</div>''', visible=True)
                 )
                 verification = result.get("verification_report", "No verification details available.")
                 logger.info(f"Verification (internal):\n{verification}")
-                # Do not display verification to user, only use internally
-                chat_history.append({"role": "user", "content": question_text})
-                chat_history.append({"role": "assistant", "content": f"**Answer:**\n{answer}"})
+                chat_history.extend([
+                    {"role": "user", "content": question_text},
+                    {"role": "assistant", "content": f"**Answer:**\n{answer}"}
+                ])
                 session_state.value["last_documents"] = retriever.invoke(question_text)
                 yield (
                     chat_history,
@@ -719,10 +712,7 @@ def main():
                     gr.update(interactive=True),
                     gr.update(interactive=True),
                     gr.update(interactive=True),
-                    gr.update(value='''<div style="background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">
-  <span id="processing-msg"></span>
-  <span id="processing-timer" style="opacity:0.8; margin-left:8px;"></span>
-</div>''', visible=True)
+                    gr.update(value='''<div style=\"background:#fff; border-radius:8px; padding:18px 24px; margin-top:32px; color:#1e293b; font-size:1.2em; font-weight:600; box-shadow:0 2px 8px rgba(0,0,0,0.04);">\n  <span id="processing-msg"></span>\n  <span id="processing-timer" style="opacity:0.8; margin-left:8px;"></span>\n</div>''', visible=True)
                 )      
                 yield (
                     chat_history,
@@ -736,8 +726,10 @@ def main():
                 )
             except Exception as e:
                 logger.error(f"Processing error: {e}", exc_info=True)
-                chat_history.append({"role": "user", "content": question_text})
-                chat_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
+                chat_history.extend([
+                    {"role": "user", "content": question_text},
+                    {"role": "assistant", "content": f"Error: {str(e)}"}
+                ])
                 yield (
                     chat_history,
                     gr.update(visible=False),
@@ -749,59 +741,20 @@ def main():
                     gr.update(value="", visible=False)
                 )
 
-        submit_btn.click(
-            fn=process_question,
-            inputs=[question, files, chat],
-            outputs=[chat, doc_context_display, refresh_context_btn, submit_btn, question, files, example_dropdown, processing_message],
-            queue=True,
-            show_progress=True
-        )
-
-        def refresh_context():
+        def refresh_context(chat_history):
             docs = session_state.value.get("last_documents", [])
             last_question = ""
-            for msg in reversed(chat.value or []):
-                if msg["role"] == "user":
-                    last_question = msg["content"]
+            for msg in reversed(chat_history or []):
+                if msg.get("role") == "user":
+                    last_question = msg.get("content", "")
                     break
             return format_document_context(docs, last_question)
 
         refresh_context_btn.click(
             fn=refresh_context,
-            inputs=[],
+            inputs=[chat],
             outputs=[doc_context_display]
         )
-
-        def load_example(example_key):
-            if not example_key or example_key not in EXAMPLES:
-                return [], "", "Select a valid example from the dropdown above"
-            ex_data = EXAMPLES[example_key]
-            question_text = ex_data["question"]
-            file_paths = ex_data["file_paths"]
-            import tempfile
-            temp_dir = tempfile.mkdtemp()
-            copied_files = []
-            file_info_text = f"Loaded: {example_key}\n\n"
-            for source_file_path in file_paths:
-                abs_source = os.path.abspath(source_file_path)
-                if os.path.exists(abs_source):
-                    filename = os.path.basename(abs_source)
-                    temp_file_path = os.path.join(temp_dir, filename)
-                    shutil.copy2(abs_source, temp_file_path)
-                    copied_files.append(temp_file_path)
-                    file_size_mb = os.path.getsize(temp_file_path) / (1024 * 1024)
-                    file_info_text += f"{filename} ({file_size_mb:.2f} MB)\n"
-                else:
-                    file_info_text += f"{source_file_path} not found\n"
-            if not copied_files:
-                return [], "", "Could not load example files"
-            return copied_files, question_text, file_info_text          
-        
-        example_dropdown.change(
-            fn=load_example,
-            inputs=[example_dropdown],
-            outputs=[files, question, loaded_file_info]
-        )
     # Launch server - Compatible with both local and Hugging Face Spaces
     # HF Spaces sets SPACE_ID environment variable
     is_hf_space = os.environ.get("SPACE_ID") is not None