Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py  —  BizIntel AI Ultra  (Gemini 1.5 Pro, advanced pipeline)
+# app.py  —  BizIntel AI Ultra  (Gemini 1.5 Pro, selectable date column)
 
 import os
 import tempfile
@@ -8,21 +8,23 @@ 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   # must accept date_col param
+from tools.forecaster import forecast_tool   # accepts date_col
 
 # ──────────────────────────────────────────────────────────────
 # 1.  GEMINI CONFIG  (1.5‑Pro, temperature 0.7)
 # ──────────────────────────────────────────────────────────────
 genai.configure(api_key=os.getenv("GEMINI_APIKEY"))
 gemini = genai.GenerativeModel(
-    model="gemini-1.5-pro-latest",
-    temperature=0.7,
-    top_p=0.9,
-    response_mime_type="text/markdown",
+    "gemini-1.5-pro-latest",
+    generation_config={
+        "temperature": 0.7,
+        "top_p": 0.9,
+        "response_mime_type": "text/markdown",
+    },
 )
 
 # ──────────────────────────────────────────────────────────────
-# 2.  STREAMLIT PAGE CONFIG
+# 2.  STREAMLIT PAGE SETUP
 # ──────────────────────────────────────────────────────────────
 st.set_page_config(page_title="BizIntel AI Ultra – Gemini 1.5 Pro", layout="wide")
 st.title("📊 BizIntel AI Ultra – Advanced Analytics Pipeline")
@@ -42,29 +44,25 @@ with open(csv_path, "wb") as f:
     f.write(csv_file.read())
 st.success("CSV saved to temporary storage ✅")
 
-# Preview first rows
+# Preview first rows and let user pick date column
 df_preview = pd.read_csv(csv_path, nrows=5)
 st.dataframe(df_preview)
-
-# ──────────────────────────────────────────────────────────────
-# 4.  DATE COLUMN SELECTION
-# ──────────────────────────────────────────────────────────────
 date_col = st.selectbox("Choose the date/time column for forecasting", df_preview.columns)
 
 # ──────────────────────────────────────────────────────────────
-# 5.  LOCAL TOOL EXECUTION
+# 4.  LOCAL TOOL EXECUTION
 # ──────────────────────────────────────────────────────────────
 with st.spinner("🔍 Parsing CSV…"):
     summary_text = parse_csv_tool(csv_path)
 
 with st.spinner("📈 Generating sales trend chart…"):
-    _ = plot_sales_tool(csv_path)   # creates sales_plot.png
+    _ = plot_sales_tool(csv_path)   # generates sales_plot.png
 
 with st.spinner("🔮 Forecasting future metrics…"):
     forecast_text = forecast_tool(csv_path, date_col=date_col)  # pass chosen column
 
 # ──────────────────────────────────────────────────────────────
-# 6.  GEMINI 1.5‑PRO STRATEGY (stream)
+# 5.  GEMINI 1.5‑PRO STRATEGY (streaming)
 # ──────────────────────────────────────────────────────────────
 prompt = (
     f"You are **BizIntel Strategist AI**.\n\n"
@@ -74,8 +72,8 @@ prompt = (
     "Return Markdown with:\n"
     "1. **Five key insights** (bullet list)\n"
     "2. **Three actionable strategies** (with expected impact)\n"
-    "3. **Risk factors or anomalies** you see\n"
-    "4. **Suggested additional visuals** to aid stakeholder understanding\n"
+    "3. **Risk factors or anomalies**\n"
+    "4. **Suggested additional visuals**\n"
 )
 
 st.subheader("🚀 Strategy Recommendations (Gemini 1.5 Pro)")
@@ -88,17 +86,13 @@ for chunk in gemini.generate_content_stream(prompt):
         placeholder.markdown(strategy_md)
 
 # ──────────────────────────────────────────────────────────────
-# 7.  DISPLAY SUMMARY & CHARTS
+# 6.  DISPLAY SUMMARY & CHARTS
 # ──────────────────────────────────────────────────────────────
 st.markdown("---")
 st.subheader("📑 CSV Summary")
 st.text(summary_text)
 
-sales_plot_path = os.path.join("sales_plot.png")
-forecast_plot_path = os.path.join("forecast_plot.png")
-
-if os.path.exists(sales_plot_path):
-    st.image(sales_plot_path, caption="Sales Trend", use_column_width=True)
-
-if os.path.exists(forecast_plot_path):
-    st.image(forecast_plot_path, caption="Forecast Chart", use_column_width=True)
+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)