Update app.py

app.py

@@ -659,6 +659,7 @@ def display_documents():
         label="Select documents to query or delete"
     )
 
+# Define initial conversation
 def initial_conversation():
     return [
         (None, "Welcome! I'm your AI assistant for web search and PDF analysis. Here's how you can use me:\n\n"
@@ -681,13 +682,13 @@ use_web_search = gr.Checkbox(label="Use Web Search", value=False)
 
 custom_placeholder = "Ask a question (Note: You can toggle between Web Search and PDF Chat in Additional Inputs below)"
 
+# Create the Gradio interface
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# AI-powered PDF Chat and Web Search Assistant")
     
     audio_input = gr.Audio(sources=["microphone"], type="numpy", label="Speak your query")
     
-    # Create the Chatbot component separately
     chatbot = gr.Chatbot(
         show_copy_button=True,
         likeable=True,
@@ -696,9 +697,9 @@ with gr.Blocks() as demo:
         value=initial_conversation()
     )
     
-    # Create the ChatInterface using the pre-defined Chatbot
     chat_interface = gr.ChatInterface(
         fn=process_input,
+        chatbot=chatbot,
         additional_inputs=[
             gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[3]),
             gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
@@ -707,11 +708,25 @@ with gr.Blocks() as demo:
             gr.CheckboxGroup(label="Select documents to query"),
             audio_input
         ],
-        chatbot=chatbot,  # Use the pre-defined Chatbot
         title="AI-powered PDF Chat and Web Search Assistant",
         description="Chat with your PDFs, use web search to answer questions, or speak your query.",
         theme=gr.themes.Soft(
-            # ... (theme configuration remains the same)
+            primary_hue="orange",
+            secondary_hue="amber",
+            neutral_hue="gray",
+            font=[gr.themes.GoogleFont("Exo"), "ui-sans-serif", "system-ui", "sans-serif"]
+        ).set(
+            body_background_fill_dark="#0c0505",
+            block_background_fill_dark="#0c0505",
+            block_border_width="1px",
+            block_title_background_fill_dark="#1b0f0f",
+            input_background_fill_dark="#140b0b",
+            button_secondary_background_fill_dark="#140b0b",
+            border_color_accent_dark="#1b0f0f",
+            border_color_primary_dark="#1b0f0f",
+            background_fill_secondary_dark="#0c0505",
+            color_accent_soft_dark="transparent",
+            code_background_fill_dark="#140b0b"
         ),
         css=css,
         examples=[
@@ -724,7 +739,7 @@ with gr.Blocks() as demo:
         analytics_enabled=False,
         textbox=gr.Textbox(placeholder="Ask a question about the uploaded PDFs or any topic", container=False, scale=7),
     )
-with demo:
+
     gr.Markdown("## Upload and Manage PDF Documents")
     with gr.Row():
         file_input = gr.Files(label="Upload your PDF documents", file_types=[".pdf"])
@@ -735,25 +750,23 @@ with demo:
     update_output = gr.Textbox(label="Update Status")
     delete_button = gr.Button("Delete Selected Documents")
     
-    # Update both the output text and the document selector
+    # Set up event handlers for document management
     update_button.click(
         update_vectors, 
         inputs=[file_input, parser_dropdown], 
-        outputs=[update_output, demo.additional_inputs[-1]]  # Use the CheckboxGroup from additional_inputs
+        outputs=[update_output, chat_interface.additional_inputs[-2]]  # Assuming the CheckboxGroup is the second-to-last additional input
     )
     
-    # Add the refresh button functionality
     refresh_button.click(
         refresh_documents, 
         inputs=[], 
-        outputs=[demo.additional_inputs[-1]]  # Use the CheckboxGroup from additional_inputs
+        outputs=[chat_interface.additional_inputs[-2]]  # Assuming the CheckboxGroup is the second-to-last additional input
     )
     
-    # Add the delete button functionality
     delete_button.click(
         delete_documents,
-        inputs=[demo.additional_inputs[-1]],  # Use the CheckboxGroup from additional_inputs
-        outputs=[update_output, demo.additional_inputs[-1]]
+        inputs=[chat_interface.additional_inputs[-2]],  # Assuming the CheckboxGroup is the second-to-last additional input
+        outputs=[update_output, chat_interface.additional_inputs[-2]]
     )
 
     gr.Markdown(
@@ -770,4 +783,4 @@ with demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)