Update app.py

app.py

@@ -243,7 +243,7 @@ def retry_last_response(history, use_web_search, model, temperature, num_calls):
     
     return chatbot_interface(last_user_msg, history, use_web_search, model, temperature, num_calls)
 
-def respond(message, history, model, temperature, num_calls, selected_docs, use_web_search):
+def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):
     logging.info(f"User Query: {message}")
     logging.info(f"Model Used: {model}")
     logging.info(f"Search Type: {'Web Search' if use_web_search else 'PDF Search'}")
@@ -478,60 +478,56 @@ def display_documents():
 
 # Define the checkbox outside the demo block
 document_selector = gr.CheckboxGroup(label="Select documents to query")
+
 use_web_search = gr.Checkbox(label="Use Web Search", value=True)
 
-demo = gr.Blocks()
+demo = gr.ChatInterface(
+    respond,
+    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"),
+        gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
+        use_web_search,
+        document_selector  # Add the document selector to the chat interface
+    ],
+    title="AI-powered Web Search and PDF Chat Assistant",
+    description="Chat with your PDFs or use web search to answer questions (Please use toggle under Additional Inputs to swithc between PDF and Web Search, Default Value Web Search)",
+    theme=gr.themes.Soft(
+        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=[
+        ["Tell me about the contents of the uploaded PDFs."],
+        ["What are the main topics discussed in the documents?"],
+        ["Can you summarize the key points from the PDFs?"]
+    ],
+    cache_examples=False,
+    analytics_enabled=False,
+)
 
+# Add file upload functionality
 with demo:
-    chatbot = gr.ChatInterface(
-        respond,
-        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"),
-            gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
-            document_selector, # Add the document selector to the chat interface
-            use_web_search
-        ],
-        title="AI-powered Web Search and PDF Chat Assistant",
-        description="Chat with your PDFs or use web search to answer questions",
-        theme=gr.themes.Soft(
-            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=[
-            ["Tell me about the contents of the uploaded PDFs."],
-            ["What are the main topics discussed in the documents?"],
-            ["Can you summarize the key points from the PDFs?"]
-        ],
-        cache_examples=False,
-        analytics_enabled=False,
-    )
-
-
-    # Add file upload functionality
     gr.Markdown("## Upload PDF Documents")
 
     with gr.Row():
         file_input = gr.Files(label="Upload your PDF documents", file_types=[".pdf"])
         parser_dropdown = gr.Dropdown(choices=["pypdf", "llamaparse"], label="Select PDF Parser", value="llamaparse")
         update_button = gr.Button("Upload Document")
-
-
     
     update_output = gr.Textbox(label="Update Status")