Update app.py

app.py

@@ -1,57 +1,78 @@
-import os, tempfile, streamlit as st, pandas as pd
+import os
+import tempfile
+import streamlit as st
 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
+# ──────────────────────────────────────────────────────────────
+#  0.  CONFIGURE GEMINI
+# ──────────────────────────────────────────────────────────────
 genai.configure(api_key=os.getenv("GEMINI_APIKEY"))
-model = genai.GenerativeModel("gemini-pro")
+gemini = genai.GenerativeModel("gemini-pro")
 
-st.set_page_config(page_title="BizIntel AI Ultra", layout="wide")
-st.title("📊 BizIntel AI Ultra – Emergency Pipeline")
+# ──────────────────────────────────────────────────────────────
+#  1.  STREAMLIT SETUP
+# ──────────────────────────────────────────────────────────────
+st.set_page_config(page_title="BizIntel AI Ultra – Lite", layout="wide")
+st.title("📊 BizIntel AI Ultra – Quick Analytics Pipeline")
 
 TEMP_DIR = tempfile.gettempdir()
 
-# 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.  CSV UPLOAD
+# ──────────────────────────────────────────────────────────────
+uploaded_csv = st.file_uploader("Upload a CSV file", type=["csv"])
+if not uploaded_csv:
+    st.info("Upload a CSV to begin.")
+    st.stop()
+
+csv_path = os.path.join(TEMP_DIR, uploaded_csv.name)
+with open(csv_path, "wb") as f:
+    f.write(uploaded_csv.read())
+st.success("CSV saved to temporary storage ✅")
 
-    # 2. Run local tools
-    summary = parse_csv_tool(csv_path)
-    plot_msg = plot_sales_tool(csv_path)
-    forecast_msg = forecast_tool(csv_path)
+# ──────────────────────────────────────────────────────────────
+#  3.  LOCAL TOOL PROCESSING
+# ──────────────────────────────────────────────────────────────
+summary_text = parse_csv_tool(csv_path)
+plot_msg      = plot_sales_tool(csv_path)    # creates sales_plot.png
+forecast_text = forecast_tool(csv_path)      # creates forecast_plot.png
 
-    # 3. Strategy via Gemini
-    prompt = f"""
-    You are a business strategist AI.
-    CSV SUMMARY:
-    {summary}
+# ──────────────────────────────────────────────────────────────
+#  4.  GEMINI STRATEGY GENERATION
+# ──────────────────────────────────────────────────────────────
+prompt = f"""
+You are BizIntel Strategist AI.
 
-    FORECAST:
-    {forecast_msg}
+CSV SUMMARY:
+{summary_text}
 
-    Provide key insights and 3 actionable strategies.
-    """
+FORECAST OUTPUT:
+{forecast_text}
 
-    strat = model.generate_content(prompt).text
+Using the data above, produce:
+1. Key insights (bullet list)
+2. Three actionable strategies to improve business performance.
 
-    # 4. Display results
-    st.subheader("📝 Data Summary")
-    st.text(summary)
+Be concise and insightful.
+"""
 
-    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)
+strategy = gemini.generate_content(prompt).text
 
-    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)
+# ──────────────────────────────────────────────────────────────
+#  5.  DISPLAY RESULTS
+# ──────────────────────────────────────────────────────���───────
+st.subheader("📑 Data Summary")
+st.text(summary_text)
 
-    st.subheader("🚀 Strategy Recommendations")
-    st.markdown(strat)
-else:
-    st.info("Upload a CSV to begin.")
+if os.path.exists("sales_plot.png"):
+    st.image("sales_plot.png", caption="Sales Trend", use_column_width=True)
+
+if os.path.exists("forecast_plot.png"):
+    st.image("forecast_plot.png", caption="Forecast Chart", use_column_width=True)
+
+st.subheader("🚀 Strategy Recommendations")
+st.markdown(strategy)