Update app.py

app.py

@@ -21,21 +21,24 @@ response_manager = ResponseManager(vector_store_id)
 # Define the chatbot function to handle user queries and generate responses
 def chat_interaction(query: str) -> str:
     """
-    Function to handle the chatbot interaction.
+    Function to handle the chatbot interaction and maintain conversation history.
     :param query: The user query to respond to.
-    :return: The response text from the chatbot.
+    :param history: The conversation history (list of [input, output] pairs).
+    :return: Updated conversation history.
     """
     try:
         if query.strip():
             response = response_manager.create_response(query, model, temperature, max_output_tokens, max_num_results)
             if not response:
-                return "Sorry, I couldn't generate a response at this time. Please try again later."
-            # Return the response from the AI model
-            return response
+                response = "Sorry, I couldn't generate a response at this time. Please try again later."
+            # Append the conversation
+            history.append((query, response))
         else:
-            return "Please enter a valid query."
+            history.append((query, "Please enter a valid query."))
+        return history
     except Exception as e:
-        return str(e)
+        history.append((query, str(e)))
+        return history
         
 # Set parameters for the response generation
 model = "gpt-4o-mini" # Set the model to be used for response generation
@@ -68,43 +71,35 @@ if not all([chatbot_title, chatbot_description,
 
 # Define the reset function
 def reset_output():
-    return chatbot_output_placeholder
+    return []
 
 # Create a Gradio Blocks interface
 with gr.Blocks() as demo:
     gr.Markdown(f"## {chatbot_title}\n{chatbot_description}")
 
-    with gr.Row():
-        user_input = gr.Textbox(
-            lines=7, 
-            label=chatbot_input_label, 
-            placeholder=chatbot_input_placeholder
-        )
-
-    with gr.Row():
-        # Output box always visible with initial placeholder
-        output = gr.Textbox(
-            lines=7,
-            label=chatbot_output_label,
-            placeholder=chatbot_output_placeholder,
-            interactive=False
-        )
+    # Chatbot history component
+    chatbot_output = gr.Chatbot(label="Chat History")
+    
+    # User input
+    user_input = gr.Textbox(
+        lines=2,
+        label=chatbot_input_label,
+        placeholder=chatbot_input_placeholder
+    )
 
-    # Place Reset on the Left and Submit on the Right
+    # Buttons
     with gr.Row():
-        # Left Column for Reset
+        # Reset button on the left (Gray)
         with gr.Column(scale=1):
             reset = gr.Button(chatbot_reset_button, variant="secondary")  
-        # Right Column for Submit
+        # Submit button on the right (Blue)
         with gr.Column(scale=1):
             submit = gr.Button(chatbot_submit_button, variant="primary")  
-            
-    # Define button click actions
-    submit.click(fn=chat_interaction, inputs=user_input, outputs=output)
-    reset.click(fn=reset_output, inputs=None, outputs=output)
 
-    # Enable "Enter" key to submit
-    user_input.submit(fn=chat_interaction, inputs=user_input, outputs=output)
+    # Submit and Reset button actions
+    submit.click(fn=chatbot, inputs=[user_input, chatbot_output], outputs=chatbot_output)
+    user_input.submit(fn=chatbot, inputs=[user_input, chatbot_output], outputs=chatbot_output)
+    reset.click(fn=reset_output, inputs=None, outputs=chatbot_output)
 
 if __name__ == "__main__":
     demo.launch()