Update app.py

app.py

@@ -101,13 +101,28 @@ def query_document(question):
 # ✅ Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 📄 AI-Powered Multi-Document Chatbot with Voice Output")
+    
     file_dropdown = gr.Dropdown(choices=get_files_from_drive(), label="📂 Select Files", multiselect=True)
+    refresh_button = gr.Button("🔄 Refresh Files")  # 🔄 Add Refresh Button
     process_button = gr.Button("🚀 Process Documents")
+    
     user_input = gr.Textbox(label="🔎 Ask a Question")
     submit_button = gr.Button("💬 Get Answer")
+    
     response_output = gr.Textbox(label="📝 Response")
     audio_output = gr.Audio(label="🔊 Audio Response")
+
+    # 🔄 Function to Refresh File List
+    def refresh_files():
+        return gr.update(choices=get_files_from_drive())
+
+    # ✅ Connect Refresh Button
+    refresh_button.click(refresh_files, outputs=file_dropdown)
+
+    # ✅ Connect Process Button
     process_button.click(process_documents, inputs=file_dropdown, outputs=response_output)
+
+    # ✅ Connect Query Button
     submit_button.click(query_document, inputs=user_input, outputs=[response_output, audio_output])
 
 demo.launch()