Update app.py

app.py

@@ -1,62 +1,78 @@
-import os 
-import langchain
-import openai
+import os
 from dotenv import load_dotenv
+from datetime import datetime
 from langchain.chat_models import ChatOpenAI
 from langchain.tools import Tool
 from langchain.utilities import SerpAPIWrapper
 from langchain.agents import initialize_agent, AgentType
 from langchain_core.exceptions import OutputParserException
-
 import gradio as gr
 
-
-
+# Load environment variables
 load_dotenv(override=True)
 
-MODEL = "gpt-4o-mini"
+MODEL = "gpt-4o"  
 
-# serpapi wrapper 
-search = SerpAPIWrapper()
+# Setup the serpapi wrapper
+search = SerpAPIWrapper(params={"num": "10", "hl": "en", "gl": "in"})
 
-# here is the defination of the tool 
+# Define tool with valid name pattern and enhanced description
 tools = [
     Tool(
-        name="Google Search",
+        name="google_search",
         func=search.run,
-        description="Always use this tool for up-to-date or current event questions. Do not guess if the answer is about ongoing events, news, or recent updates."
+        description="Use this tool ONLY to fetch real-time data or current events. Always add today's date to ensure accuracy. Never rely on prior knowledge or make assumptions about ongoing events."
     )
 ]
 
-#define llm with temperature 0
-llm = ChatOpenAI(temperature=0.7, model=MODEL) 
-# define an agent using athe   tool that we have defined above
-agent = initialize_agent(tools, llm, agent=AgentType.OPENAI_FUNCTIONS, handle_parsing_errors=True, verbose=True)
+# Initialize the LLM
+llm = ChatOpenAI(temperature=0, model=MODEL)
+
+# Initialize agent with OPENAI_FUNCTIONS mode
+agent = initialize_agent(
+    tools,
+    llm,
+    agent=AgentType.OPENAI_FUNCTIONS,
+    handle_parsing_errors=True,
+    verbose=True
+)
+
+# Function to inject current date into query if missing
+def add_current_date_to_query(query):
+    today = datetime.now().strftime("%B %Y")
+    if today.lower() not in query.lower():
+        query = f"{query} ({today})"
+    return query
 
-# chat bot definition with history 
+# Chatbot function with exception handling and enforced date
 def chatbot(user_input):
     if user_input.lower().strip() in ["hi", "hello", "hey"]:
-        return "Hello! How can I help you today?"
+        return "Hello! How can I assist you today?"
     try:
-        # Example agent usage (replace with your real agent.run)
-        output = agent.run(user_input)
+        enhanced_input = add_current_date_to_query(user_input)
+        output = agent.run(enhanced_input)
     except OutputParserException:
         output = "Agent encountered a parsing issue. Please rephrase your query."
     except Exception as e:
         output = f"Unexpected Error: {e}"
     return output
-# UI USING GRADIO 
+
+# Gradio UI with chat history
 with gr.Blocks() as demo:
     chatbot_ui = gr.Chatbot(type="messages")
-    user_input = gr.Textbox(placeholder="Ask Anything...")
+    user_input = gr.Textbox(placeholder="Ask about latest events, news, or anything...")
     submit = gr.Button("Ask")
+
     def respond(user_message, chat_history):
         chat_history = chat_history or []
         chat_history.append({"role": "user", "content": user_message})
         response = chatbot(user_message)
         chat_history.append({"role": "assistant", "content": response})
         return chat_history, ""
+
     user_input.submit(respond, inputs=[user_input, chatbot_ui], outputs=[chatbot_ui, user_input])
     submit.click(respond, inputs=[user_input, chatbot_ui], outputs=[chatbot_ui, user_input])
 
+# Launch app
 demo.launch(inline=False)
+