Update app.py

app.py

@@ -7,11 +7,35 @@ import asyncio
 from gtts import gTTS
 from typing_extensions import TypedDict
 from langgraph.graph import StateGraph, START, END
+import csv
 
 # Load API keys
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
+# --- Initialize Database from external CSV ---
+def init_db_from_csv(csv_path: str = "transactions.csv") -> None:
+    """Create 'transactions' table and load data from the provided CSV file."""
+    conn = sqlite3.connect("shop.db")
+    cur = conn.cursor()
+    cur.execute(
+        "CREATE TABLE IF NOT EXISTS transactions (date TEXT, product TEXT, amount REAL)"
+    )
+    with open(csv_path, newline='') as f:
+        reader = csv.DictReader(f)
+        rows = [(row["date"], row["product"], float(row["amount"])) for row in reader]
+    # Replace old data
+    cur.execute("DELETE FROM transactions")
+    cur.executemany(
+        "INSERT INTO transactions (date, product, amount) VALUES (?, ?, ?)", rows
+    )
+    conn.commit()
+    conn.close()
+
+# Initialize DB at startup (ensure transactions.csv is present)
+init_db_from_csv()
+
 # --- Business Logic Functions ---
+
 def db_agent(query: str) -> str:
     try:
         conn = sqlite3.connect("shop.db")
@@ -31,7 +55,8 @@ def db_agent(query: str) -> str:
             return f"Top product today: {row[0]} with ₹{row[1]:,.2f}"
         return "No transactions found for today."
     except sqlite3.OperationalError as e:
-        return f"Database error: {e}. Please initialize 'transactions' table in shop.db."
+        return f"Database error: {e}. Please check 'transactions' table in shop.db."
+
 
 def web_search_agent(query: str) -> str:
     try:
@@ -46,6 +71,7 @@ def web_search_agent(query: str) -> str:
         pass
     return llm_agent(query)
 
+
 def llm_agent(query: str) -> str:
     response = openai.chat.completions.create(
         model="gpt-4o-mini",
@@ -57,6 +83,7 @@ def llm_agent(query: str) -> str:
     )
     return response.choices[0].message.content.strip()
 
+
 def stt_agent(audio_path: str) -> str:
     with open(audio_path, "rb") as afile:
         transcript = openai.audio.transcriptions.create(
@@ -65,6 +92,7 @@ def stt_agent(audio_path: str) -> str:
         )
     return transcript.text.strip()
 
+
 def tts_agent(text: str, lang: str = 'en') -> str:
     tts = gTTS(text=text, lang=lang)
     out_path = "response_audio.mp3"
@@ -77,6 +105,7 @@ class State(TypedDict):
     result: str
 
 # Routing logic based on query
+
 def route_fn(state: State) -> str:
     q = state["query"].lower()
     if any(k in q for k in ["max revenue", "revenue"]):
@@ -135,7 +164,7 @@ def handle_query(audio_or_text: str):
 # --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("## Shop Voice-Box Assistant (Speech In/Out)")
-    inp = gr.Audio(sources=["microphone"], type="filepath", label="Speak or type your question")
+    inp = gr.Audio(sources=["microphone"], type="filepath", label="Speak or type your question or upload transactions.csv separately in root")
     out_text = gr.Textbox(label="Answer (text)")
     out_audio = gr.Audio(label="Answer (speech)")
     submit = gr.Button("Submit")