Update app.py

app.py

@@ -157,49 +157,65 @@ p {
 }
 """
 
+def upload_pdf(pdf_file):
+    global vectorstore, chain  # Use global variables to store state
 
-# Create the Gradio interface
-with gr.Blocks(css=custom_css) as demo:
-    pdf_input = gr.File(label="ارففع ملف PDF")
-    chat_input = gr.Textbox(label="أدخل سؤالك هنا")
-    chat_output = gr.Textbox(label="الرد الآلي")
-    audio_output = gr.Audio(label="استمع إلى الرد")
-    submit_button = gr.Button("إرسال")
     data = load_pdf(pdf_file)
     vectorstore = prepare_vectorstore(data)
+    chain = create_chain(vectorstore)
 
-    # Define the logic for processing the PDF and generating responses
-    def process_pdf_and_chat(pdf_file, user_input):
-        
-        chain = create_chain(vectorstore)
+    return "PDF uploaded and processed successfully!"
 
-        prompt = f"""
-        You are an expert Arabic-language assistant specialized in analyzing and responding to queries about Arabic PDF documents. Your responses should be precise, informative, and reflect the professional tone and structure expected in formal Arabic communication. Focus on extracting and presenting relevant information from the document clearly and systematically, while avoiding colloquial or informal language.
+def chat(user_input):
+    global chain  # Access the global chain variable
+
+    if chain is None:
+        return "Please upload a PDF file first.", None  # Prompt user to upload a PDF
 
+    prompt = f"""
+        You are an expert Arabic-language assistant specialized in analyzing and responding to queries about Arabic PDF documents. Your responses should be precise, informative, and reflect the professional tone and structure expected in formal Arabic communication. Focus on extracting and presenting relevant information from the document clearly and systematically, while avoiding colloquial or informal language.
         When responding, ensure the following:
            - Your answer directly reflects the content of the document.
            - If the requested information is not available in the document, clearly state that.
            - Keep your response concise yet comprehensive, addressing the question fully.
            - Always respond in formal Arabic, without using English.\n
-
         Question: {user_input}\n
         Helpful Answer:"""
 
-        response = chain({"question": prompt})
-        assistant_response = response["answer"]
+    response = chain({"question": prompt})
+    assistant_response = response["answer"]
+
+    # Generate a unique identifier for the audio file
+    audio_id = str(uuid.uuid4())
 
-        # Generate a unique identifier for the audio file
-        audio_id = str(uuid.uuid4())
+    # Create audio file
+    tts = gTTS(text=assistant_response, lang='ar')
+    audio_file = f"{audio_id}.mp3"
+    tts.save(audio_file)
 
-        # Create audio file
-        tts = gTTS(text=assistant_response, lang='ar')
-        audio_file = f"{audio_id}.mp3"
-        tts.save(audio_file)
+    return assistant_response, audio_file
+
+with gr.Blocks(css=custom_css) as demo:
+    
+    pdf_input = gr.File(label="اختر ملف PDF")
+    
+    submit_button_pdf = gr.Button("ارفع الملف")
+    
+    chat_input = gr.Textbox(label="أدخل سؤالك هنا")
+    
+    chat_output = gr.Textbox(label="الرد الآلي")
+    
+    audio_output = gr.Audio(label="استمع إلى الرد")
+    
+    submit_button_chat = gr.Button("إرسال")
 
-        return assistant_response, audio_file
+    
+    # Connect upload button to upload function
+    submit_button_pdf.click(upload_pdf, inputs=[pdf_input], outputs="text")
 
-    # Connect the button to the processing function
-    submit_button.click(process_pdf_and_chat, inputs=[pdf_input, chat_input], outputs=[chat_output, audio_output])
+    
+    # Connect chat button to chat function    
+    submit_button_chat.click(chat, inputs=[chat_input], outputs=[chat_output, audio_output])
 
 # Launch the Gradio app
 demo.launch()