Update app.py

app.py

@@ -476,41 +476,22 @@ def display_documents():
         label="Select documents to query"
     )
 
-def chat_with_parameters(message, history, model, temperature, num_calls, use_web_search, selected_docs):
-    """
-    This function wraps the chatbot functionality with all parameters.
-    It will be used as the main function for gr.Interface.
-    """
-    try:
-        for response in respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):
-            history.append((message, response))
-            yield history
-    except Exception as e:
-        history.append((message, f"An error occurred: {str(e)}"))
-        yield history
+# Define the checkbox outside the demo block
+document_selector = gr.CheckboxGroup(label="Select documents to query")
 
-# Define the components that were previously in additional_inputs
-model_dropdown = gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[3])
-temperature_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature")
-api_calls_slider = gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls")
 use_web_search = gr.Checkbox(label="Use Web Search", value=True)
-document_selector = gr.CheckboxGroup(label="Select documents to query")
 
-# Create the main interface
-demo = gr.Interface(
-    fn=chat_with_parameters,
-    inputs=[
-        gr.Textbox(lines=2, label="Chat Input"),
-        "state",
-        model_dropdown,
-        temperature_slider,
-        api_calls_slider,
+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
+        document_selector  # Add the document selector to the chat interface
     ],
-    outputs=gr.Chatbot(),
     title="AI-powered Web Search and PDF Chat Assistant",
-    description="Chat with your PDFs or use web search to answer questions",
+    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",