Upload 8 files

.gitignore

@@ -0,0 +1,50 @@
+# Byte-compiled / optimized / compiled files
+__pycache__/
+*.pyc
+*.pyd
+*.pyo
+*.egg-info/
+.pytest_cache/
+
+# Virtual environment
+.venv/
+env/
+venv/
+*.egg
+*.whl
+
+# Editor and IDE configurations
+.vscode/
+.idea/
+.DS_Store
+
+# Environment variables
+.env
+.env.*
+# For example: .env.local, .env.development, .env.production
+
+# Database files
+*.sqlite3
+*.db
+# Add specific database files if applicable, e.g., for SQLite:
+# instance/*.sqlite
+
+# Logs and temporary files
+*.log
+tmp/
+logs/
+
+# Test reports
+.coverage
+htmlcov/
+
+# Build artifacts
+dist/
+build/
+
+# Uploaded media (if applicable)
+# media/
+# uploads/
+
+# Specific to FastAPI/Uvicorn
+.uvicorn_pid

agent.py

@@ -0,0 +1,13 @@
+# Import the pre-built agent creator
+from langgraph.prebuilt import create_react_agent
+
+# Import your model and tools from other files
+from llm import model
+from tools import all_tools
+
+from prompts import SYSTEM_PROMPT as system_prompt
+
+# Create the agent using the model, tools, and the new system prompt
+app = create_react_agent(model=model, tools=all_tools, prompt=system_prompt)
+
+print("Pre-built ReAct agent created successfully!")

app.py

@@ -0,0 +1,50 @@
+import requests
+import gradio as gr
+import uuid
+
+# The URL of your running FastAPI backend
+BACKEND_URL = "http://127.0.0.1:8000/invoke"
+
+# Generate a unique session ID for the entire chat session
+SESSION_ID = str(uuid.uuid4())
+
+print(f"New chat session started with ID: {SESSION_ID}")
+
+def get_agent_response(message, history):
+    """
+    This function is called every time the user sends a message.
+    It calls your FastAPI backend and returns the agent's response.
+    """
+    # The payload to send to your backend
+    payload = {
+        "query": message,
+        "session_id": SESSION_ID
+    }
+    
+    try:
+        # Make the API call to your backend
+        response = requests.post(BACKEND_URL, json=payload)
+        response.raise_for_status()  # Raise an exception for bad status codes (4xx or 5xx)
+        
+        # Extract the agent's response text from the JSON
+        return response.json()["response"]
+        
+    except requests.exceptions.RequestException as e:
+        print(f"Error calling backend: {e}")
+        return "Sorry, there was an error connecting to the agent. Please check the backend server."
+
+# Create the Gradio Chat Interface
+chatbot_ui = gr.ChatInterface(
+    fn=get_agent_response,
+    title="Sarvam AI Agent",
+    description="Ask me about horoscopes, holidays, and more!",
+    examples=[
+        "What is the horoscope for Leo today?", 
+        "Tell me the panchang for today",
+        "What are the Indian holidays in 2025?"
+        ]
+)
+
+# Launch the UI
+if __name__ == "__main__":
+    chatbot_ui.launch()

llm.py

@@ -0,0 +1,19 @@
+import os
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+
+load_dotenv()
+
+SARVAM_API_KEY = os.getenv("SARVAM_API_KEY")
+
+if not SARVAM_API_KEY:
+    raise ValueError("SARVAM_API_KEY not found in .env file. Please make sure it's set.")
+
+model = ChatOpenAI(
+    model="sarvam-m",
+    api_key=SARVAM_API_KEY,
+    base_url="https://api.sarvam.ai/v1",
+    temperature=0.2,
+)
+
+print("Sarvam AI model initialized successfully using ChatOpenAI client.")

main.py

@@ -0,0 +1,60 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import List, Tuple
+from langchain_core.messages import AIMessage
+from agent import app
+from datetime import datetime
+
+api = FastAPI(
+    title="Sarvam AI Agent Server (ReAct)",
+    description="An API for interacting with the LangGraph pre-built ReAct agent.",
+    version="2.0.0"
+)
+
+class QueryRequest(BaseModel):
+    query: str
+    session_id: str
+
+sessions: dict[str, List[Tuple[str, str]]] = {}
+
+def get_current_date_context():
+    """Gets the current date and formats it as a string for the AI."""
+    now = datetime.now()
+    return f"For context, today's date is {now.strftime('%A, %B %d, %Y')}."
+
+@api.post("/invoke")
+def invoke_agent(request: QueryRequest):
+    """
+    Main endpoint to run the agent. Manages conversation history.
+    """
+    current_messages = sessions.get(request.session_id, [])
+
+    if not current_messages:
+        query_with_context = f"{request.query} ({get_current_date_context()})"
+        user_message = ("user", query_with_context)
+    else:
+        user_message = ("user", request.query)
+
+    current_messages.append(user_message)
+    
+    inputs = {"messages": current_messages}
+    
+    final_response_content = ""
+    print("\n--- AGENT STREAM STARTED ---")
+    for event in app.stream(inputs, stream_mode="values"):
+        messages = event.get('messages', [])
+        if messages:
+            final_message = messages[-1]
+            
+            # --- DEBUG PRINT STATEMENT ---
+            # This will show us every step the agent takes.
+            print(f"DEBUG: {final_message}\n")
+
+            if isinstance(final_message, AIMessage) and not final_message.tool_calls:
+                final_response_content = final_message.content
+
+    print("--- AGENT STREAM ENDED ---\n")
+    current_messages.append(("assistant", final_response_content))
+    sessions[request.session_id] = current_messages
+
+    return {"response": final_response_content}

prompts.py

@@ -0,0 +1,38 @@
+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. 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.
+"""

tools.py

@@ -0,0 +1,168 @@
+import datetime
+import json
+import requests
+import pytz
+from dateutil import parser
+from typing import Optional, Literal
+from langchain.tools import tool
+
+
+@tool
+def get_horoscope(sign: str, date: str = None, language: str = "EN") -> str:
+    """
+    Fetches the horoscope for a given zodiac sign and date from the ExaWeb API.
+    Defaults to today if no date is provided.
+
+    Args:
+        sign: Zodiac sign (e.g., Aries, Taurus, Gemini).
+        date: Date in any recognizable format (optional).
+        language: Language code (e.g., 'EN' for English, 'HI' for Hindi).
+    """
+    try:
+        if date:
+            date_obj = parser.parse(date)
+        else:
+            date_obj = datetime.datetime.now()
+        
+        formatted_date = date_obj.strftime("%d-%m-%Y")
+        
+        params = {
+            "rashi": sign.upper(),
+            "language": language,
+            "day": formatted_date
+        }
+        url = "https://api.exaweb.in:3004/api/rashi"
+
+        response = requests.get(url, params=params)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+        
+        data = response.json()
+        sign_data = data.get(sign.upper())
+
+        if sign_data:
+            return json.dumps(sign_data)
+        
+        return f"INFO: No horoscope found for sign: {sign}"
+
+    except requests.exceptions.RequestException as e:
+        return f"ERROR: [get_horoscope]: Network error while fetching horoscope: {e}"
+    except Exception as e:
+        return f"ERROR: [get_horoscope]: An unexpected error occurred: {str(e)}"
+
+
+
+@tool
+def get_date_panchang(date: str = None, data_language: str = "EN") -> str:
+    """
+    Fetches the Panchang data for a given date.
+    If the user does not provide a date, use today's real date.
+
+    Args:
+        date: Date in any format (optional).
+        data_language: Language of the Panchang data (e.g., 'EN' for English, 'HI' for Hindi).
+    """
+    try:
+        if not date:
+            now = datetime.datetime.now()
+        else:
+            now = parser.parse(date)
+        
+        api_date = now.strftime("%d/%m/%y")
+        
+        url = (
+            f"https://api.exaweb.in:3004/api/panchang/daily?"
+            f"date={api_date}&app_language=EN&data_language={data_language}"
+        )
+        headers = {"api_key": "anvl_bharat_cal123"}
+        
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        data = response.json()
+
+        if not isinstance(data, dict) or not data:
+            return "ERROR: [get_date_panchang]: Received empty or invalid data from API."
+
+        # Simplified formatting for clarity
+        return json.dumps(data)
+
+    except requests.exceptions.RequestException as e:
+        return f"ERROR: [get_date_panchang]: Network error while fetching Panchang: {e}"
+    except Exception as e:
+        return f"ERROR: [get_date_panchang]: An unexpected error occurred: {str(e)}"
+
+
+
+@tool
+def get_holidays(year: int = None, data_language: str = "EN") -> str:
+    """
+    Fetches holidays from all categories for a given year from ExaWeb API.
+
+    Args:
+        year: Year (e.g., 2025). Optional — defaults to current year.
+        data_language: Data language for holidays (default: "EN").
+    """
+    if not year:
+        year = datetime.datetime.now().year
+        
+    params = {"data_language": data_language, "year": year}
+    headers = {"api_key": "anvl_bharat_cal123"}
+
+    try:
+        response = requests.get(
+            "https://api.exaweb.in:3004/api/panchang/holiday",
+            params=params,
+            headers=headers
+        )
+        response.raise_for_status()
+        data = response.json()
+        return json.dumps(data)
+
+    except requests.exceptions.RequestException as e:
+        return f"ERROR: [get_holidays]: Network error fetching holidays: {e}"
+    except Exception as e:
+        return f"ERROR: [get_holidays]: An unexpected error occurred: {str(e)}"
+
+
+
+@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 from the ExaWeb API.
+    
+    Args:
+        year: The year to fetch (e.g., 2025). Defaults to current year.
+        month: The full month name to fetch (e.g., "june"). Defaults to current month.
+        data_language: The language for the festival names (default: "EN").
+    """
+    if not year:
+        year = datetime.datetime.now().year
+    if not month:
+        month = datetime.datetime.now().strftime("%B").lower()
+    else:
+        month = month.lower()
+
+    api_url = "https://api.exaweb.in:3004/api/panchang/festival"
+    params = {"year": year, "month": month, "data_language": data_language}
+    headers = {"api_key": "anvl_bharat_cal123"}
+
+    try:
+        response = requests.get(api_url, params=params, headers=headers)
+        response.raise_for_status()
+        data = response.json()
+        return json.dumps(data)
+
+    except requests.exceptions.RequestException as e:
+        return f"ERROR: [get_monthly_festivals]: Network error while fetching data: {e}"
+    except Exception as e:
+        return f"ERROR: [get_monthly_festivals]: An unexpected error occurred: {str(e)}"
+
+
+# List of all tools available in the module
+
+all_tools = [
+    get_horoscope, 
+    get_date_panchang, 
+    get_holidays, 
+    get_monthly_festivals
+]
+