import gradio as gr
from gradio_client import Client

# Call the existing model
client = Client("Futuresony/Mr.Events")

# 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")
    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

    # 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])

# Launch the Gradio interface
demo.launch(debug=True, show_error=True) # Corrected launch call and added show_error