Update app.py

app.py

@@ -412,6 +412,14 @@ def process_voice_note(audio_file, history):
         history.append({"role": "assistant", "content": "It seems there was an error while transcribing audio due to a technical issue. If there's anything specific from the document or any other questions you have regarding the content, please let me know, and I can assist you with that information."})
         return "", history, "", None, None
 
+def reset_all():
+    rag.thread_id = None
+    if hasattr(rag, 'file_ids'):
+        rag.file_ids = []
+    if hasattr(rag, 'vector_store_id'):
+        rag.vector_store_id = None
+    return "", [], "", None, None
+
 # Create Gradio interface with improved layout
 with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
     gr.Markdown("""
@@ -425,34 +433,33 @@ with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
     audio_input = gr.Audio(type="filepath", label="Record or Upload Audio", elem_classes="gradio-audio", visible=False)
     tts_output = gr.Audio(label="Assistant Voice Reply", interactive=False, visible=False)
 
-    with gr.Group(elem_classes="compact-box"):
-        gr.Markdown("<div class='section-title'>Document Q&A</div>")
-        with gr.Row():
-            with gr.Column(scale=1, min_width=350):
+    with gr.Row():
+        # Left: Document Q&A controls
+        with gr.Column(scale=1, min_width=350):
+            with gr.Group(elem_classes="compact-box"):
+                gr.Markdown("<div class='section-title'>Document Q&A Controls</div>")
                 file_input = gr.File(label="Upload Document", file_types=[".pdf", ".txt", ".doc", ".docx"], file_count="single", type="binary", elem_classes="upload-btn")
                 mic_btn = gr.Button("🎤 Record Voice", elem_classes="audio-btn")
                 audio_input
                 send_voice_btn = gr.Button("Send Voice Note", elem_classes="send-btn", visible=False)
                 reset_btn = gr.Button("Reset Chat & Upload New Document", elem_classes="reset-btn")
                 file_output
-            with gr.Column(scale=3, min_width=500):
+                question
+                file_input.change(process_file, file_input, file_output)
+                reset_btn.click(reset_all, None, [file_output, chatbot, question, audio_input, tts_output])
+                def show_audio():
+                    return {audio_input: gr.update(visible=True), send_voice_btn: gr.update(visible=True)}
+                mic_btn.click(show_audio, None, [audio_input, send_voice_btn])
+                def hide_audio():
+                    return {audio_input: gr.update(visible=False), send_voice_btn: gr.update(visible=False)}
+                send_voice_btn.click(process_voice_note, [audio_input, chatbot], [file_output, chatbot, question, audio_input, tts_output])
+                send_voice_btn.click(hide_audio, None, [audio_input, send_voice_btn])
+                question.submit(process_question, [question, chatbot], [question, chatbot, question, audio_input])
+                tts_output
+        # Right: Chatbot screen
+        with gr.Column(scale=2, min_width=500):
+            with gr.Group(elem_classes="compact-box"):
                 chatbot
-        with gr.Row():
-            question
-        file_input.change(process_file, file_input, file_output)
-        def reset_all():
-            rag.thread_id = None
-            return "", [], "", None, None
-        reset_btn.click(reset_all, None, [file_output, chatbot, question, audio_input, tts_output])
-        def show_audio():
-            return {audio_input: gr.update(visible=True), send_voice_btn: gr.update(visible=True)}
-        mic_btn.click(show_audio, None, [audio_input, send_voice_btn])
-        def hide_audio():
-            return {audio_input: gr.update(visible=False), send_voice_btn: gr.update(visible=False)}
-        send_voice_btn.click(process_voice_note, [audio_input, chatbot], [file_output, chatbot, question, audio_input, tts_output])
-        send_voice_btn.click(hide_audio, None, [audio_input, send_voice_btn])
-        question.submit(process_question, [question, chatbot], [question, chatbot, question, audio_input])
-        tts_output
 
     # Add JavaScript for audio handling
     demo.load(