Spaces:
Running
Running
| import gradio as gr | |
| import json | |
| import os | |
| from smol import ChatAgent | |
| GLOBAL_STATE_FILE = "global_state.json" | |
| if os.path.exists(GLOBAL_STATE_FILE): | |
| with open(GLOBAL_STATE_FILE, "r") as f: | |
| global_game_state = json.load(f) | |
| else: | |
| global_game_state = {} # initialize an empty state | |
| def process_chat(message, chat_history): | |
| if chat_history is None: | |
| chat_history = [] | |
| chat_history.append(("User", message)) | |
| response = ChatAgent(message) | |
| chat_history.append(("AI", response)) | |
| return "", chat_history | |
| def chat_function(user_prompt, history): | |
| # Initialize history if empty | |
| if history is None: | |
| history = [] | |
| # Append the user's message as a dictionary | |
| history.append({"role": "user", "content": user_prompt}) | |
| # Process the prompt using your ChatAgent to get the AI response | |
| ai_response = ChatAgent(user_prompt) | |
| print(f"AI RESPONSE: {ai_response}") | |
| print("HISTORY: ", history) | |
| # Append the AI's response as a dictionary | |
| history.append({"role": "assistant", "content": ai_response}) | |
| # Clear the input and return the updated history | |
| return "", history | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## Live Game & Chat Interface") | |
| with gr.Row(): | |
| # Left Column: Iframe for the game | |
| with gr.Column(scale=3): | |
| gr.Markdown("### Game") | |
| # The iframe points to your static Game | |
| gr.HTML( | |
| value=""" | |
| <iframe src="https://experiment-sandpack.vercel.app" style="width:100%; height:600px; border:none;"></iframe> | |
| """ | |
| ) | |
| # Right Column: Chat interface (smaller column) | |
| with gr.Column(scale=1): | |
| gr.Markdown("### Chat") | |
| chatbot = gr.Chatbot(type="messages",label="Conversation") | |
| # Textbox to receive user prompt | |
| txt_input = gr.Textbox(placeholder="Type your prompt here...", label="Your Message") | |
| # State to hold the conversation history | |
| state = gr.State([]) | |
| # When the user submits a message, update the chat | |
| txt_input.submit(chat_function, inputs=[txt_input, state], outputs=[txt_input, chatbot]) | |
| # chat_output = gr.Chatbot(label="Chat Output", type="messages", interactive=False) | |
| # chat_input = gr.Textbox( | |
| # placeholder="Type your message here...", | |
| # label="Your Message" | |
| # ) | |
| # chat_history = gr.State([]) | |
| # chat_input.submit( | |
| # process_chat, | |
| # inputs=[chat_input, chat_history], | |
| # outputs=[chat_input, chat_output], | |
| # ) | |
| demo.launch() | |