Update app.py

app.py

@@ -1,61 +1,57 @@
-import os, tempfile, streamlit as st
-from google.adk.runner import run_agent_sync     # ← NEW
-from agents.analytics_pipeline import analytics_coordinator
-from db_connector import fetch_data_from_db, list_tables, SUPPORTED_ENGINES
+import os, tempfile, streamlit as st, pandas as pd
+import google.generativeai as genai
+from tools.csv_parser import parse_csv_tool
+from tools.plot_generator import plot_sales_tool
+from tools.forecaster import forecast_tool
+
+# Configure Gemini
+import os
+genai.configure(api_key=os.getenv("GEMINI_APIKEY"))
+model = genai.GenerativeModel("gemini-pro")
 
 st.set_page_config(page_title="BizIntel AI Ultra", layout="wide")
-st.title("📊 BizIntel AI Ultra – Business Intelligence Agent")
+st.title("📊 BizIntel AI Ultra – Emergency Pipeline")
 
 TEMP_DIR = tempfile.gettempdir()
 
-# ── Data source ───────────────────────────────────────────────
-src = st.radio("Select data source", ["Upload CSV", "Connect to SQL Database"])
-csv_path = None
+# 1. CSV upload
+csv = st.file_uploader("Upload CSV", ["csv"])
+if csv:
+    csv_path = os.path.join(TEMP_DIR, csv.name)
+    with open(csv_path, "wb") as f:
+        f.write(csv.read())
+    st.success("CSV saved ✅")
+
+    # 2. Run local tools
+    summary = parse_csv_tool(csv_path)
+    plot_msg = plot_sales_tool(csv_path)
+    forecast_msg = forecast_tool(csv_path)
+
+    # 3. Strategy via Gemini
+    prompt = f"""
+    You are a business strategist AI.
+    CSV SUMMARY:
+    {summary}
+
+    FORECAST:
+    {forecast_msg}
+
+    Provide key insights and 3 actionable strategies.
+    """
+
+    strat = model.generate_content(prompt).text
+
+    # 4. Display results
+    st.subheader("📝 Data Summary")
+    st.text(summary)
+
+    if "Generated sales_plot.png" in plot_msg and os.path.exists("sales_plot.png"):
+        st.image("sales_plot.png", caption="Sales Trend", use_column_width=True)
 
-if src == "Upload CSV":
-    up = st.file_uploader("Upload CSV", ["csv"])
-    if up:
-        csv_path = os.path.join(TEMP_DIR, up.name)
-        with open(csv_path, "wb") as f:
-            f.write(up.read())
-        st.success("CSV saved ✅")
+    if "Generated forecast_plot.png" in forecast_msg and os.path.exists("forecast_plot.png"):
+        st.image("forecast_plot.png", caption="Forecast Chart", use_column_width=True)
 
+    st.subheader("🚀 Strategy Recommendations")
+    st.markdown(strat)
 else:
-    eng = st.selectbox("DB engine", SUPPORTED_ENGINES)
-    conn = st.text_input("SQLAlchemy connection string")
-    if conn:
-        tbls = list_tables(conn)
-        tbl = st.selectbox("Table", tbls) if tbls else None
-        if tbl:
-            csv_path = fetch_data_from_db(conn, tbl)
-            st.success(f"Fetched **{tbl}** as CSV ✅")
-
-# ── Optional image preview ────────────────────────────────────
-st.markdown("---")
-img = st.file_uploader("Optional image (PNG/JPG)", ["png", "jpg", "jpeg"])
-if img:
-    img_path = os.path.join(TEMP_DIR, img.name)
-    with open(img_path, "wb") as f:
-        f.write(img.read())
-    st.image(img_path, use_column_width=True)
-
-# ── Run pipeline via ADK runner ───────────────────────────────
-if csv_path:
-    st.markdown("---")
-    st.info("Running analytics pipeline…")
-
-    try:
-        report = run_agent_sync(agent=analytics_coordinator, input_data=csv_path)
-    except Exception as e:
-        st.error(f"Agent execution failed: {e}")
-        st.stop()
-
-    st.subheader("📝 Analysis & Strategy Report")
-    st.text(report)
-
-    # Show charts if generated
-    for name, cap in [("sales_plot.png", "Sales Trend"),
-                      ("forecast_plot.png", "Forecast Chart")]:
-        p = os.path.join(TEMP_DIR, name)
-        if os.path.exists(p):
-            st.image(p, caption=cap, use_column_width=True)
+    st.info("Upload a CSV to begin.")