Update app.py

app.py

@@ -413,6 +413,9 @@ with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
     # Define shared components at the top
     chatbot = gr.Chatbot(height=350, elem_classes="gradio-chatbot", label=None, type="messages")
     file_output = gr.Textbox(label="Upload Status", interactive=False, elem_classes="textbox")
+    question = gr.Textbox(label="Type your question and press Enter", placeholder="Ask a question about your document...", elem_classes="textbox")
+    audio_input = gr.Audio(type="filepath", label="Record or Upload Audio", elem_classes="gradio-audio", visible=False)
+    audio_status = gr.Textbox(label="Audio Status", interactive=False, elem_classes="status-text", visible=False)
 
     # Document upload section
     with gr.Row(elem_classes="centered-main"):
@@ -423,9 +426,9 @@ with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
                 reset_btn = gr.Button("Reset Chat & Upload New Document", elem_classes="reset-btn")
                 def reset_all():
                     rag.thread_id = None
-                    return "", []
+                    return "", [], "", None  # file_output, chatbot, question, audio_input
                 file_input.change(process_file, file_input, file_output)
-                reset_btn.click(reset_all, None, [file_output, chatbot])
+                reset_btn.click(reset_all, None, [file_output, chatbot, question, audio_input])
 
     # Unified chat and voice panel
     with gr.Row(elem_classes="centered-main"):
@@ -433,16 +436,13 @@ with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
             with gr.Group(elem_classes="compact-box"):
                 gr.Markdown("<div class='section-title'>2. Chat with your Document</div>")
                 with gr.Row():
-                    question = gr.Textbox(label="Type your question and press Enter", placeholder="Ask a question about your document...", elem_classes="textbox")
                     send_btn = gr.Button("Send", elem_classes="send-btn")
                     mic_btn = gr.Button("🎤", elem_classes="audio-btn")
-                question.submit(process_question, [question, chatbot], [question, chatbot])
-                send_btn.click(process_question, [question, chatbot], [question, chatbot])
+                question.submit(process_question, [question, chatbot], [question, chatbot, question, audio_input])
+                send_btn.click(process_question, [question, chatbot], [question, chatbot, question, audio_input])
 
                 # Inline audio recorder, hidden by default
-                audio_input = gr.Audio(type="filepath", label="Record or Upload Audio", elem_classes="gradio-audio", visible=False)
-                audio_status = gr.Textbox(label="Audio Status", interactive=False, elem_classes="status-text", visible=False)
-                audio_input.change(process_audio, [audio_input, chatbot], [audio_input, chatbot])
+                audio_input.change(process_audio, [audio_input, chatbot], [audio_input, chatbot, question, audio_input])
 
                 def show_audio():
                     return {audio_input: gr.update(visible=True), audio_status: gr.update(visible=True)}