Update app.py

app.py

@@ -8,7 +8,7 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import tools_condition
 from langgraph.checkpoint.memory import MemorySaver
-from tools import search_tool, weather_info_tool, hub_stats_tool
+from tools import search_tool, hub_stats_tool, weather_info_tool
 from retriever import guest_info_tool
 import gradio as gr
 
@@ -19,14 +19,13 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 #llm = ChatGroq(model="qwen-2.5-coder-32b")
 llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash")
 
-tools = [search_tool, weather_info_tool, hub_stats_tool, guest_info_tool]
+tools = [search_tool, hub_stats_tool, guest_info_tool, weather_info_tool]
 chat_with_tools = llm.bind_tools(tools)
 
 # System message
 sys_msg = SystemMessage(content="""
 Role:
 You are a helpful agent and hosting a party.
-
 STRICT RULES:
 1. Follow a THINK → TOOL → THINK → RESPOND approach:
     - THINK: Analyze the request and decide if any tool call is required or if it can be answered without a tool.
@@ -41,9 +40,9 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
 def assistant(state: AgentState):
-    return {
-        "messages": [chat_with_tools.invoke(state["messages"])],
-    }
+    if len(state["messages"]) > 7:
+        return {"messages": chat_with_tools.invoke([sys_msg] + state["messages"][-6:])}
+    return {"messages": chat_with_tools.invoke([sys_msg] + state["messages"])}
 
 ## The graph
 builder = StateGraph(AgentState)
@@ -78,14 +77,12 @@ def alfred_response(question):
 
 # Gradio
 
-input_textbox = gr.Textbox(label="Type your query here:", placeholder="Hi", lines=5)
-output_textbox = gr.Textbox(label="Type your query here:", placeholder="Hi", lines=5)
 gr.Interface(
     fn=alfred_response, 
     inputs="text", 
     outputs="text", 
-    title="Party Organizer Helper", 
+    title="Party Organizer Assistant", 
     description="Helps you answer with different asks during Party",
     examples=[["Whats weather now in Bangalore?"], ["The weather in Bangalore is Rainy with a temperature of 15°C."]],
     live=True
-).launch(share=False)
+).launch()