Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py  —  BizIntel AI Ultra  (Gemini 1.5 Pro, selectable date column)
+# app.py  —  BizIntel AI Ultra  (Gemini 1.5 Pro, date‑column selector)
 
 import os
 import tempfile
@@ -8,7 +8,7 @@ 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   # accepts date_col
+from tools.forecaster import forecast_tool   # accepts date_col argument
 
 # ──────────────────────────────────────────────────────────────
 # 1.  GEMINI CONFIG  (1.5‑Pro, temperature 0.7)
@@ -36,7 +36,7 @@ TEMP_DIR = tempfile.gettempdir()
 # ──────────────────────────────────────────────────────────────
 csv_file = st.file_uploader("Upload CSV (≤ 200 MB)", type=["csv"])
 if csv_file is None:
-    st.info("⬆️  Upload a CSV to begin.")
+    st.info("⬆️ Upload a CSV to begin.")
     st.stop()
 
 csv_path = os.path.join(TEMP_DIR, csv_file.name)
@@ -44,10 +44,10 @@ with open(csv_path, "wb") as f:
     f.write(csv_file.read())
 st.success("CSV saved to temporary storage ✅")
 
-# Preview first rows and let user pick date column
+# Preview first rows and choose date column
 df_preview = pd.read_csv(csv_path, nrows=5)
 st.dataframe(df_preview)
-date_col = st.selectbox("Choose the date/time column for forecasting", df_preview.columns)
+date_col = st.selectbox("Select the date/time column for forecasting", df_preview.columns)
 
 # ──────────────────────────────────────────────────────────────
 # 4.  LOCAL TOOL EXECUTION
@@ -56,20 +56,20 @@ with st.spinner("🔍 Parsing CSV…"):
     summary_text = parse_csv_tool(csv_path)
 
 with st.spinner("📈 Generating sales trend chart…"):
-    _ = plot_sales_tool(csv_path)   # generates sales_plot.png
+    _ = plot_sales_tool(csv_path)   # saves sales_plot.png
 
 with st.spinner("🔮 Forecasting future metrics…"):
-    forecast_text = forecast_tool(csv_path, date_col=date_col)  # pass chosen column
+    forecast_text = forecast_tool(csv_path, date_col=date_col)  # uses chosen column
 
 # ──────────────────────────────────────────────────────────────
-# 5.  GEMINI 1.5‑PRO STRATEGY (streaming)
+# 5.  GEMINI 1.5‑PRO STRATEGY (sync call)
 # ──────────────────────────────────────────────────────────────
 prompt = (
     f"You are **BizIntel Strategist AI**.\n\n"
     f"## CSV Summary\n```\n{summary_text}\n```\n\n"
     f"## Forecast Output\n```\n{forecast_text}\n```\n\n"
     "### Task\n"
-    "Return Markdown with:\n"
+    "Return **Markdown** with:\n"
     "1. **Five key insights** (bullet list)\n"
     "2. **Three actionable strategies** (with expected impact)\n"
     "3. **Risk factors or anomalies**\n"
@@ -77,13 +77,11 @@ prompt = (
 )
 
 st.subheader("🚀 Strategy Recommendations (Gemini 1.5 Pro)")
-placeholder = st.empty()
-strategy_md = ""
+with st.spinner("Gemini 1.5 Pro is thinking…"):
+    response = gemini.generate_content(prompt)
 
-for chunk in gemini.generate_content_stream(prompt):
-    if chunk.candidate and chunk.candidate.content.parts:
-        strategy_md += chunk.candidate.content.parts[0].text
-        placeholder.markdown(strategy_md)
+strategy_md = response.text
+st.markdown(strategy_md)
 
 # ──────────────────────────────────────────────────────────────
 # 6.  DISPLAY SUMMARY & CHARTS
@@ -94,5 +92,6 @@ 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)