Update app.py

app.py

@@ -1,78 +1,72 @@
+# app.py  –  BizIntel AI Ultra (Gemini 1.5 Pro, selectable date column)
+
 import os
 import tempfile
+import pandas as pd
 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
+from tools.forecaster import forecast_tool  # must accept date_col param
 
 # ──────────────────────────────────────────────────────────────
-#  0.  CONFIGURE GEMINI
+# 1.  GEMINI CONFIG  (1.5‑pro, temperature 0.7)
 # ──────────────────────────────────────────────────────────────
 genai.configure(api_key=os.getenv("GEMINI_APIKEY"))
-gemini = genai.GenerativeModel("gemini-pro")
+gemini = genai.GenerativeModel(
+    model="gemini-1.5-pro-latest",
+    temperature=0.7,
+    top_p=0.9,
+    response_mime_type="text/markdown",
+)
 
 # ──────────────────────────────────────────────────────────────
-#  1.  STREAMLIT SETUP
+# 2.  STREAMLIT PAGE SETUP
 # ──────────────────────────────────────────────────────────────
-st.set_page_config(page_title="BizIntel AI Ultra – Lite", layout="wide")
-st.title("📊 BizIntel AI Ultra – Quick Analytics Pipeline")
+st.set_page_config(page_title="BizIntel AI Ultra – Gemini 1.5 Pro", layout="wide")
+st.title("📊 BizIntel AI Ultra – Advanced Analytics Pipeline")
 
 TEMP_DIR = tempfile.gettempdir()
 
 # ──────────────────────────────────────────────────────────────
-#  2.  CSV UPLOAD
+# 3.  CSV UPLOAD
 # ──────────────────────────────────────────────────────────────
-uploaded_csv = st.file_uploader("Upload a CSV file", type=["csv"])
-if not uploaded_csv:
-    st.info("Upload a CSV to begin.")
+csv_file = st.file_uploader("Upload a CSV file (≤ 200 MB)", type=["csv"])
+if not csv_file:
+    st.info("⬆️  Upload a CSV to begin.")
     st.stop()
 
-csv_path = os.path.join(TEMP_DIR, uploaded_csv.name)
+csv_path = os.path.join(TEMP_DIR, csv_file.name)
 with open(csv_path, "wb") as f:
-    f.write(uploaded_csv.read())
+    f.write(csv_file.read())
 st.success("CSV saved to temporary storage ✅")
 
+# Preview first rows for column selection
+df_preview = pd.read_csv(csv_path, nrows=5)
+st.dataframe(df_preview)
+
 # ──────────────────────────────────────────────────────────────
-#  3.  LOCAL TOOL PROCESSING
+# 4.  DATE COLUMN CHOICE
 # ──────────────────────────────────────────────────────────────
-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
+date_col = st.selectbox("Choose the date/time column for forecasting", df_preview.columns)
 
 # ──────────────────────────────────────────────────────────────
-#  4.  GEMINI STRATEGY GENERATION
+# 5.  RUN LOCAL TOOLS
 # ──────────────────────────────────────────────────────────────
-prompt = f"""
-You are BizIntel Strategist AI.
-
-CSV SUMMARY:
-{summary_text}
-
-FORECAST OUTPUT:
-{forecast_text}
+with st.spinner("Parsing CSV…"):
+    summary_text = parse_csv_tool(csv_path)
 
-Using the data above, produce:
-1. Key insights (bullet list)
-2. Three actionable strategies to improve business performance.
+with st.spinner("Generating sales trend chart…"):
+    trend_msg = plot_sales_tool(csv_path)  # creates sales_plot.png
 
-Be concise and insightful.
-"""
-
-strategy = gemini.generate_content(prompt).text
+with st.spinner("Forecasting future metrics…"):
+    forecast_text = forecast_tool(csv_path, date_col=date_col)  # ← passes chosen column
 
 # ──────────────────────────────────────────────────────────────
-#  5.  DISPLAY RESULTS
+# 6.  GEMINI 1.5‑PRO STRATEGY (stream)
 # ──────────────────────────────────────────────────────────────
-st.subheader("📑 Data Summary")
-st.text(summary_text)
-
-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)
+prompt = f"""
+You are BizIntel Strategist AI.
 
-st.subheader("🚀 Strategy Recommendations")
-st.markdown(strategy)
+**CSV SUMMARY**