replaced app.py

Signed-off-by: apple muncy <[email protected]>

app.py

@@ -6,31 +6,19 @@ from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import tools_condition
 from langchain_ollama.chat_models import ChatOllama
 
-from tools import search_tool, weather_info_tool, hub_stats_tool
-
 
+from tools import search_tool, weather_info_tool, hub_stats_tool
 import gradio as gr
-
-
-from smolagents import GradioUI, CodeAgent, HfApiModel
-
-
+from smolagents import GradioUI, LiteLLMModel
 from retriever import load_guest_dataset
-
-from dotenv import load_dotenv
 import yaml
-load_dotenv()
-
-# Import our custom tools from their modules
-from retriever import load_guest_dataset
+from dotenv import load_dotenv
 
+load_dotenv()
 
 # Import our custom tools from their modules
 from retriever import load_guest_dataset
 
-model = HfApiModel()
-guest_info_tool = load_guest_dataset()
-
 # Load prompts from YAML file
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -39,7 +27,7 @@ with open("prompts.yaml", 'r') as stream:
 system_prompt = prompt_templates["system_prompt"]
 
 # Initialize the chat model
-chat = ChatOllama(model="qwen2:7b",
+chat = ChatOllama(model="qwen2:7b", 
                 verbose=True)
 
 # Define available tools
@@ -52,6 +40,7 @@ tools = [
 
 # Bind tools to the chat model
 chat_with_tools = chat.bind_tools(tools)
+
 # Generate the AgentState and Agent graph
 class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
@@ -63,18 +52,17 @@ def assistant(state: AgentState):
         messages = [SystemMessage(content=system_prompt)] + state["messages"]
     else:
         messages = state["messages"]
-
+    
     return {
         "messages": [chat_with_tools.invoke(messages)],
     }
 
-
 ## The graph
 builder = StateGraph(AgentState)
 
 # Define nodes: these do the work
 builder.add_node("assistant", assistant)
-builder.add_node("tools",ToolNode(tools))
+builder.add_node("tools", ToolNode(tools))
 
 # Define edges: these determine how the control flow moves
 builder.add_edge(START, "assistant")
@@ -102,7 +90,7 @@ def chat_fn(message, history):
     history.append((message, response))
     return history, ""
 
- 
+
 # Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("### LangGraph Chat with Gradio")
@@ -113,6 +101,5 @@ with gr.Blocks() as demo:
     # Hook the send button
     send_btn.click(fn=chat_fn, inputs=[msg, chatbot], outputs=[chatbot, msg])
 
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()