Update app.py

app.py

@@ -9,22 +9,28 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # --- Agents ---
 def db_agent(query: str) -> str:
-    conn = sqlite3.connect("shop.db")
-    cur = conn.cursor()
-    if "max revenue" in query.lower():
-        cur.execute(
-            """
-            SELECT product, SUM(amount) AS revenue
-            FROM transactions
-            WHERE date = date('now')
-            GROUP BY product
-            ORDER BY revenue DESC
-            LIMIT 1
-            """
-        )
-        row = cur.fetchone()
-        return f"Top product today: {row[0]} with ₹{row[1]:,.2f}" if row else "No transactions found for today."
-    return None
+    try:
+        conn = sqlite3.connect("shop.db")
+        cur = conn.cursor()
+        if "max revenue" in query.lower():
+            cur.execute(
+                """
+                SELECT product, SUM(amount) AS revenue
+                FROM transactions
+                WHERE date = date('now')
+                GROUP BY product
+                ORDER BY revenue DESC
+                LIMIT 1
+                """
+            )
+            row = cur.fetchone()
+            if row:
+                return f"Top product today: {row[0]} with ₹{row[1]:,.2f}"
+            return "No transactions found for today."
+        return None
+    except sqlite3.OperationalError as e:
+        # Handle missing table or other DB errors
+        return f"Database error: {e}. Please initialize 'transactions' table in shop.db."
 
 
 def web_search_agent(query: str) -> str:
@@ -32,21 +38,23 @@ def web_search_agent(query: str) -> str:
     resp = requests.get(
         "https://serpapi.com/search",
         params={"q": query, "api_key": os.getenv("SERPAPI_KEY")}  
-    ).json()
-    snippet = resp.get("organic_results", [{}])[0].get("snippet", "")
+    )
+    data = resp.json()
+    snippet = data.get("organic_results", [{}])[0].get("snippet", "")
     return llm_agent(f"Summarize: {snippet}")
 
 
 def llm_agent(prompt: str) -> str:
-    resp = openai.ChatCompletion.create(
+    # Updated for openai>=1.0.0 interface
+    response = openai.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": prompt}
+            {"role": "user", "content": prompt},
         ],
-        temperature=0.2
+        temperature=0.2,
     )
-    return resp.choices[0].message.content.strip()
+    return response.choices[0].message.content.strip()
 
 
 def handle_query(query: str) -> str:
@@ -77,4 +85,4 @@ with gr.Blocks() as demo:
     submit_btn.click(fn=handle_query, inputs=user_input, outputs=response_box)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)