import random
import gradio as gr
from gradio_client import Client

def random_response(message, history):
    # Creating a persistent chat session with history for each user chat.
    client = Client("TejAndrewsACC/Z3taACC-Plus")
    
    # Make API call to chat with message and previous history
    result = client.predict(
        message=message,
        history=history,
        max_tokens=2048,
        temperature=0.7,
        top_p=0.95,
        api_name="/chat"
    )
    
    # Return the response and updated history
    return result, history + [(message, result)]  # Append the current message and response to history

# Set up Gradio chat interface
demo = gr.ChatInterface(
    fn=random_response, 
    type="messages", 
    autofocus=False, 
    save_history=True, 
    show_progress="full",
    theme="TejAndrewsACC/zetaofficialthemeacc"
)

if __name__ == "__main__":
    demo.launch()