Update app.py

app.py

@@ -4,23 +4,55 @@ from gradio_client import Client
 # Call the existing model
 client = Client("Futuresony/Mr.Events")
 
-def chat_with_model(user_input):
-    result = client.predict(
-        query=user_input,
-        api_name="/chat"
-    )
-    return result
-
-# Create a better desktop-friendly interface
+# Function to interact with the hosted model
+def chat_with_model(user_input, chat_history):
+    """
+    Sends user input and chat history to the hosted model and returns the response
+    and updated history.
+    """
+    # The hosted Gradio app expects the chat history as a list of [user, assistant] pairs.
+    # Initial call will have chat_history as an empty list.
+    # Subsequent calls will have chat_history including previous turns.
+
+    print(f"Client sending query: {user_input}")
+    print(f"Client sending history: {chat_history}")
+
+    try:
+        # Call the hosted model's chat endpoint
+        # Pass user_input and chat_history as positional arguments
+        result = client.predict(
+            user_input,
+            chat_history, # Pass the history from the client's Chatbot
+            api_name="/chat"
+        )
+        print(f"Client received raw result: {result}")
+
+        # The hosted app's `chat` function returns the final response string.
+        # We need to append the user input and the model's response to the history.
+        updated_history = chat_history + [[user_input, result]]
+        return "", updated_history # Return empty string for textbox and updated history
+
+    except Exception as e:
+        print(f"Error during client prediction: {e}")
+        import traceback
+        print(traceback.format_exc())
+        # Append user input and an error message to history
+        error_message = f"An error occurred while communicating with the model: {e}"
+        updated_history = chat_history + [[user_input, error_message]]
+        return "", updated_history
+
+
+# Create the desktop-friendly interface
 with gr.Blocks() as demo:
     gr.Markdown("## 💬 Test the ABSA Model Chat")
-    with gr.Row():
-        with gr.Column(scale=3):
-            input_text = gr.Textbox(label="Type your message")
-            submit_btn = gr.Button("Send")
-        with gr.Column(scale=5):
-            output_text = gr.Textbox(label="Model Response", lines=6)
+    chatbot = gr.Chatbot(height=400) # Chatbot to display conversation
+    msg = gr.Textbox(label="Type your message") # Textbox for user input
+    clear = gr.ClearButton([msg, chatbot]) # Button to clear
 
-    submit_btn.click(fn=chat_with_model, inputs=input_text, outputs=output_text)
+    # Link the input, button, and chatbot
+    # The fn will receive the textbox value and the chatbot history.
+    # It will return an empty string for the textbox and the updated history for the chatbot.
+    msg.submit(chat_with_model, [msg, chatbot], [msg, chatbot])
 
-demo.launch(demo.launch(debug=True, show_error=True))
+# Launch the Gradio interface
+demo.launch(debug=True, show_error=True) # Corrected launch call and added show_error