import gradio as gr
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI
from langgraph.graph import StateGraph, END
from typing import Dict, List

# Assume you have your LangGraph setup here
def agent_node(state):
    prompt = ChatPromptTemplate.from_messages(
        [("user", "{user_input}")]
    )
    model = ChatOpenAI()
    response = model.invoke(prompt.format_messages(**state))
    return {"response": response.content}

def create_workflow():
    workflow = StateGraph(Dict)
    workflow.add_node("agent", agent_node)
    workflow.set_entry_point("agent")
    workflow.add_edge("agent", END)
    return workflow.compile()

workflow = create_workflow()

def respond(message, history):
    result = workflow.invoke({"user_input": message})
    return result["response"]

iface = gr.ChatInterface(respond)
iface.launch()

gr.close_all()