Spaces:
Sleeping
Sleeping
File size: 1,859 Bytes
a505c5f b22d605 a505c5f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import gradio as gr
from modules.data_class import DataState
from modules.tools import data_node
from modules.nodes import chatbot_with_tools, human_node, maybe_exit_human_node, maybe_route_to_tools
from langgraph.graph import StateGraph, START, END
from IPython.display import Image, display
from pprint import pprint
from typing import Literal
from langgraph.prebuilt import ToolNode
from collections.abc import Iterable
from IPython.display import display, clear_output
import sys
# Define the LangGraph chatbot
graph_builder = StateGraph(DataState)
# Add nodes
graph_builder.add_node("chatbot_healthassistant", chatbot_with_tools)
graph_builder.add_node("patient", human_node)
graph_builder.add_node("documenting", data_node)
# Define edges
graph_builder.add_conditional_edges("chatbot_healthassistant", maybe_route_to_tools)
graph_builder.add_conditional_edges("patient", maybe_exit_human_node)
graph_builder.add_edge("documenting", "chatbot_healthassistant")
graph_builder.add_edge(START, "chatbot_healthassistant")
# Compile the graph
graph_with_order_tools = graph_builder.compile()
# Function to handle conversation
def chat_interface(user_input, history):
if not history:
history = []
# Create the initial state
state = DataState(messages=history, data={}, finished=False)
# Run the chatbot
for output in graph_with_order_tools.stream(state):
response = output["messages"][-1] # Extract the latest chatbot response
history.append(("User: " + user_input, "Bot: " + response))
return "", history # Return empty input and updated chat history
# Launch Gradio UI
iface = gr.ChatInterface(
fn=chat_interface,
title="LangGraph Chatbot",
description="A chatbot powered by LangGraph and hosted on Hugging Face.",
theme="compact"
)
if __name__ == "__main__":
iface.launch() |