Update app.py

app.py

@@ -205,32 +205,49 @@ with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("# ديمو بوت للقاء مركز حضرموت للدراسات التاريخية")
     gr.Markdown("## المنعقد السبت 14 - سبتمبر 2024")
     
+    # File input and process button
     with gr.Row():
         pdf_input = gr.File(label="اختر ملف PDF للدردشة")
         process_button = gr.Button("رفع وبدء الدردشة")
     
-        chat_interface = gr.ChatInterface(
-            chat,
-            title="الدردشة مع البوت",
-            description="اسأل أي سؤال عن محتوى الملف PDF",
-            theme="soft",
-            examples=["ما هو موضوع الوثيقة؟", "من هم الأشخاص المذكورون؟", "ما هي التواريخ الرئيسية المذكورة؟"],
-            cache_examples=True,
-            retry_btn=None,
-            undo_btn="مسح آخر رسالة",
-            clear_btn="مسح المحادثة",
-        )
+    # Chat interface and audio output
+    chat_interface = gr.ChatInterface(
+        chat,
+        title="الدردشة مع البوت",
+        description="اسأل أي سؤال عن محتوى الملف PDF",
+        theme="soft",
+        examples=["ما هو موضوع الوثيقة؟", "من هم الأشخاص المذكورون؟", "ما هي التواريخ الرئيسية المذكورة؟"],
+        cache_examples=True,
+        retry_btn=None,
+        undo_btn="مسح آخر رسالة",
+        clear_btn="مسح المحادثة",
+    )
     
-        audio_output = gr.Audio(label="الرد الصوتي")
-        
-        process_button.click(process_pdf, inputs=[pdf_input], outputs=[chat_interface.textbox])
+    audio_output = gr.Audio(label="الرد الصوتي")
     
-        # Use the internal Textbox and Chatbot provided by the ChatInterface
-        chat_interface.submit(
-            fn=lambda x, y: y[-1][1],
-            inputs=[chat_interface.textbox, chat_interface.chatbot],
-            outputs=[audio_output]
-        )
+    # State to store the vectorstore
+    vectorstore_state = gr.State()
+
+    # Ensure chat interface is disabled until PDF is processed
+    process_button.click(
+        fn=process_pdf, 
+        inputs=[pdf_input], 
+        outputs=[chat_interface.textbox, vectorstore_state]  # Store the vectorstore in the state
+    )
+
+    # Enable chat only after PDF is processed and vectorstore is ready
+    def handle_chat(user_input, history, vectorstore):
+        if vectorstore is None:
+            return "Please upload and process a PDF first.", ""
+        return chat(user_input, history, vectorstore)
+
+    # Use the state to pass the vectorstore to the chat
+    chat_interface.submit(
+        fn=handle_chat,
+        inputs=[chat_interface.textbox, chat_interface.chatbot, vectorstore_state],  # Pass the vectorstore as input
+        outputs=[audio_output]
+    )
 
 demo.launch()
 
+