Update app.py

app.py

@@ -19,12 +19,11 @@ from langchain_openai import ChatOpenAI
 from langgraph.prebuilt import create_react_agent
 
 # ==============================================================================
-# PART 1: TOOL DEFINITIONS (from tools.py)
+# PART 1: TOOL DEFINITIONS
 # ==============================================================================
 @tool
 def get_horoscope(sign: str, date: str = None, language: str = "EN") -> str:
     """Fetches the horoscope for a given zodiac sign and date."""
-    # ... (Your full get_horoscope function code)
     try:
         if date: date_obj = parser.parse(date)
         else: date_obj = datetime.datetime.now()
@@ -39,7 +38,6 @@ def get_horoscope(sign: str, date: str = None, language: str = "EN") -> str:
 @tool
 def get_date_panchang(date: str = None, data_language: str = "EN") -> str:
     """Fetches the Panchang data for a given date."""
-    # ... (Your full get_date_panchang function code)
     try:
         if not date: now = datetime.datetime.now()
         else: now = parser.parse(date)
@@ -54,7 +52,6 @@ def get_date_panchang(date: str = None, data_language: str = "EN") -> str:
 @tool
 def get_holidays(year: int = None, data_language: str = "EN") -> str:
     """Fetches holidays for a given year."""
-    # ... (Your full get_holidays function code)
     try:
         if not year: year = datetime.datetime.now().year
         params = {"data_language": data_language, "year": year}
@@ -67,7 +64,6 @@ def get_holidays(year: int = None, data_language: str = "EN") -> str:
 @tool
 def get_monthly_festivals(year: Optional[int] = None, month: Optional[str] = None, data_language: str = "EN") -> str:
     """Fetches festival data for a specific month and year."""
-    # ... (Your full get_monthly_festivals function code)
     try:
         if not year: year = datetime.datetime.now().year
         if not month: month = datetime.datetime.now().strftime("%B").lower()
@@ -83,56 +79,51 @@ def get_monthly_festivals(year: Optional[int] = None, month: Optional[str] = Non
 all_tools = [get_horoscope, get_date_panchang, get_holidays, get_monthly_festivals]
 
 # ==============================================================================
-# PART 2: SYSTEM PROMPT (from prompt.py)
+# PART 2: SYSTEM PROMPT
 # ==============================================================================
-SYSTEM_PROMPT = """ Your primary function is to use the provided tools to answer user queries accurately.
-    **Core Directives:**
-    1.  **Tool First:** You MUST use the provided tools to find information. Never answer from your own general knowledge. If the tools do not provide an answer, state that the information could not be found.
-    2.  **Language Match:** You MUST respond in the exact language of the user's query. You are proficient in English (EN), Hindi (HI), Bengali (BN), Gujarati (GU), Tamil (TA), Telugu (TE), Kannada (KN), Malayalam (ML), Marathi (MR), Oriya (OR), and Panjabi (PA).
-    3.  **Interpret, Don't Dump:** Your job is to interpret the JSON data returned by the tools and present it to the user in a clear, well-formatted, and human-readable way. Do not just output the raw JSON.
-    
-    ---
-    
-    **Tool Usage and Data Interpretation Guide:**
-    
-    **1. `get_horoscope`**
-       - **When to Use:** Use this tool when a user asks for a horoscope for any zodiac sign (e.g., Aries, Leo, Gemini).
-       - **Data Interpretation:** The tool returns a JSON object with keys like `prediction`, `monetary_gains`, `love_life`, `health`, `lucky_number`, and `lucky_color`. Format your response using clear headings for each of these categories. For example:
-         "Here is the horoscope for [Sign]:
-         **Prediction:** [prediction text]
-         **Love Life:** [love life text]
-         **Lucky Number:** [number]"
-    
-    **2. `get_date_panchang`**
-       - **When to Use:** Use this tool when a user asks for the "Panchang," "Panchangam," or detailed daily astrological details for a specific date.
-       - **Data Interpretation:** This tool returns a very large JSON object. **Do not dump the entire object.**
-         - If the user asks for the general Panchang, summarize the most important elements: **Sunrise, Sunset, Tithi, Nakshatra, Yoga, and Karana**.
-         - If the user asks for a specific detail (e.g., "What is Rahu Kalam today?"), find that specific key in the JSON (`Rahu Kalam`) and provide only that information.
-    
-    **3. `get_holidays`**
-       - **When to Use:** Use this tool for general queries about holidays within a specific **year**. This tool provides a list of Hindu, Islamic, Christian, and Government holidays.
-       - **Data Interpretation:** Present the holidays in a clean list format. You can group them by month if the list is long.
-    
-    **4. `get_monthly_festivals`**
-       - **When to Use:** Prefer this tool when a user asks for festivals in a specific **month**. It provides more detail than `get_holidays`.
-       - **Data Interpretation:** Format the response as a list of festivals for that month, including the date for each.
-    
-    ---
-    
-    **Final Response Style:**
-    Your final answer to the user should always be friendly, well-formatted, and directly address their question using only the data you retrieved from the tools.
-    """
+SYSTEM_PROMPT = """You are an expert astrological and calendar assistant. Your primary function is to use the provided tools to answer user queries accurately.
+
+**Core Directives:**
+1.  **Tool First:** You MUST use the provided tools to find information. Never answer from your own general knowledge. If the tools do not provide an answer, state that the information could not be found.
+2.  **Language Match:** You MUST respond in the exact language of the user's query. You are proficient in English (EN), Hindi (HI), Bengali (BN), Gujarati (GU), Tamil (TA), Telugu (TE), Kannada (KN), Malayalam (ML), Marathi (MR), Oriya (OR), and Panjabi (PA).
+3.  **Interpret, Don't Dump:** Your job is to interpret the JSON data returned by the tools and present it to the user in a clear, well-formatted, and human-readable way. Do not just output the raw JSON.
+
+---
+
+**Tool Usage and Data Interpretation Guide:**
+
+**1. `get_horoscope`**
+   - **When to Use:** Use this tool when a user asks for a horoscope for any zodiac sign (e.g., Aries, Leo, Gemini).
+   - **Data Interpretation:** The tool returns a JSON object with keys like `prediction`, `monetary_gains`, `love_life`, `health`, `lucky_number`, and `lucky_color`. Format your response using clear headings for each of these categories.
+
+**2. `get_date_panchang`**
+   - **When to Use:** Use this tool when a user asks for the "Panchang," "Panchangam," or detailed daily astrological details for a specific date.
+   - **Data Interpretation:** This tool returns a very large JSON object. **Do not dump the entire object.**
+     - If the user asks for the general Panchang, summarize the most important elements: **Sunrise, Sunset, Tithi, Nakshatra, Yoga, and Karana**.
+     - If the user asks for a specific detail (e.g., "What is Rahu Kalam today?"), find that specific key in the JSON (`Rahu Kalam`) and provide only that information.
+
+**3. `get_holidays`**
+   - **When to Use:** Use this tool for general queries about holidays within a specific **year**.
+   - **Data Interpretation:** Present the holidays in a clean list format.
+
+**4. `get_monthly_festivals`**
+   - **When to Use:** Prefer this tool when a user asks for festivals in a specific **month**. It provides more detail than `get_holidays`.
+   - **Data Interpretation:** Format the response as a list of festivals for that month, including the date for each.
+
+---
+
+**Final Response Style:**
+Your final answer to the user should always be friendly, well-formatted, and directly address their question using only the data you retrieved from the tools.
+"""
 
 # ==============================================================================
-# PART 3: MODEL INITIALIZATION (from llm.py)
+# PART 3: MODEL INITIALIZATION
 # ==============================================================================
 load_dotenv()
 SARVAM_API_KEY = os.getenv("SARVAM_API_KEY")
 
 if not SARVAM_API_KEY:
     print("Warning: SARVAM_API_KEY secret not found.")
-    # In a deployed space, we can't raise an error, so we'll let it fail later
-    # raise ValueError("SARVAM_API_KEY not found. Please set it in the Space secrets.")
 
 model = ChatOpenAI(
     model="sarvam-m",
@@ -143,9 +134,9 @@ model = ChatOpenAI(
 print("Sarvam AI model initialized successfully.")
 
 # ==============================================================================
-# PART 4: AGENT CREATION (from agent.py)
+# PART 4: AGENT CREATION
 # ==============================================================================
-agent_app = create_react_agent(model=model, tools=all_tools, prompt=SYSTEM_PROMPT)
+agent_app = create_react_agent(model=model, tools=all_tools, system_message=SYSTEM_PROMPT)
 print("Pre-built ReAct agent created successfully.")
 
 # ==============================================================================
@@ -179,12 +170,8 @@ chatbot_ui = gr.ChatInterface(
     fn=get_agent_response,
     title="Jyotish AI Assistant",
     description="Ask about horoscopes, panchang, holidays, and festivals.",
-    type="messages",
-    examples=[
-        "What is the horoscope for Leo today?",
-        "Tell me the panchang for today",
-        "What are the Indian holidays in 2025?"
-    ]
+    type="messages"
+    # The 'examples' line has been removed to prevent the startup crash.
 )
 
 # Launch the Gradio app