Update app.py

app.py

@@ -412,6 +412,8 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
     logging.info(f"Selected Documents: {selected_docs}")
     logging.info(f"Use Web Search: {use_web_search}")
 
+    history = history + [(message, "")]
+
     if use_web_search:
         original_query = message
         rephrased_query = rephrase_query(message, conversation_manager)
@@ -475,7 +477,15 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
                 fallback_model = "mistralai/Mistral-7B-Instruct-v0.3"
                 yield from respond(message, history, fallback_model, temperature, num_calls, selected_docs)
             else:
-                yield f"An error occurred with the {model} model: {str(e)}. Please try again or select a different model."
+                yield f"An error occurred with the {model} model: {str(e)}. Please try again or select a different model
+
+            history[-1] = (message, response)
+            return response, history
+        except Exception as e:
+            logging.error(f"Error in respond: {str(e)}")
+            error_message = f"An error occurred: {str(e)}"
+            history[-1] = (message, error_message)
+            return error_message, history
         
 logging.basicConfig(level=logging.DEBUG)
 
@@ -730,8 +740,16 @@ demo = gr.Interface(
 
 # Add file upload functionality
 # Add file upload functionality
-with demo:
-    gr.Markdown("## Chat Interface")
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(
+        show_copy_button=True,
+        likeable=True,
+        layout="bubble",
+        height=400,
+        value=initial_conversation()
+    )
+    state = gr.State([])
+
     with gr.Row():
         text_input = gr.Textbox(
             placeholder=custom_placeholder,
@@ -739,16 +757,32 @@ with demo:
             scale=7
         )
         audio_input = gr.Audio(source="microphone", type="filepath")
-        submit_button = gr.Button("Submit")
 
-    chat_output = gr.Chatbot(
-        show_copy_button=True,
-        likeable=True,
-        layout="bubble",
-        height=400,
-        value=initial_conversation()
+    with gr.Accordion("⚙️ Parameters", open=False):
+        model = gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[3])
+        temperature = gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature")
+        num_calls = 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)
+        selected_docs = gr.CheckboxGroup(label="Select documents to query")
+
+    submit_button = gr.Button("Submit")
+
+    submit_button.click(
+        fn=respond,
+        inputs=[
+            text_input,
+            state,
+            model,
+            temperature,
+            num_calls,
+            use_web_search,
+            selected_docs,
+            audio_input
+        ],
+        outputs=[chatbot, state]
     )
-with demo:
+
+    # Add file upload functionality
     gr.Markdown("## Upload and Manage PDF Documents")
     with gr.Row():
         file_input = gr.Files(label="Upload your PDF documents", file_types=[".pdf"])
@@ -759,25 +793,22 @@ 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
     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, selected_docs]
     )
     
-    # Add the refresh button functionality
     refresh_button.click(
         refresh_documents, 
         inputs=[], 
-        outputs=[demo.additional_inputs[-1]]  # Use the CheckboxGroup from additional_inputs
+        outputs=[selected_docs]
     )
     
-    # 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=[selected_docs],
+        outputs=[update_output, selected_docs]
     )
 
     gr.Markdown(
@@ -786,12 +817,11 @@ with demo:
     1. Upload PDF documents using the file input at the top.
     2. Select the PDF parser (pypdf or llamaparse) and click "Upload Document" to update the vector store.
     3. Select the documents you want to query using the checkboxes.
-    4. Ask questions in the chat interface. 
+    4. Ask questions by typing in the text box or using the microphone for speech input.
     5. Toggle "Use Web Search" to switch between PDF chat and web search.
     6. Adjust Temperature and Number of API Calls to fine-tune the response generation.
     7. Use the provided examples or ask your own questions.
     """
     )
-
 if __name__ == "__main__":
     demo.launch(share=True)