Update app.py

app.py

@@ -1,24 +1,31 @@
+import os
+import tempfile
 import streamlit as st
 from agents.analytics_pipeline import analytics_coordinator
-import os
 from db_connector import fetch_data_from_db, list_tables, SUPPORTED_ENGINES
 
+# ──────────────────────────────────────────────────────────────
+#  0.  PAGE CONFIG
+# ──────────────────────────────────────────────────────────────
 st.set_page_config(page_title="BizIntel AI Ultra", layout="wide")
 st.title("📊 BizIntel AI Ultra – Business Intelligence Agent")
 
+# Where we'll store any uploaded files (always writable in HF Spaces)
+TEMP_DIR = tempfile.gettempdir()
+
 # ──────────────────────────────────────────────────────────────
-# ❶  DATA SOURCE SELECTION (CSV or Database)
+#  1.  DATA SOURCE SELECTION
 # ──────────────────────────────────────────────────────────────
 input_source = st.radio("Select data source", ["Upload CSV", "Connect to SQL Database"])
-file_path = None
+csv_path: str | None = None
 
 if input_source == "Upload CSV":
     uploaded_file = st.file_uploader("Upload CSV", type="csv")
     if uploaded_file:
-        file_path = os.path.join("data", uploaded_file.name)
-        with open(file_path, "wb") as f:
+        csv_path = os.path.join(TEMP_DIR, uploaded_file.name)
+        with open(csv_path, "wb") as f:
             f.write(uploaded_file.read())
-        st.success("CSV file uploaded ✅")
+        st.success("CSV file saved to temporary storage ✅")
 
 elif input_source == "Connect to SQL Database":
     engine = st.selectbox("Database engine", SUPPORTED_ENGINES)
@@ -28,41 +35,43 @@ elif input_source == "Connect to SQL Database":
         if tables:
             table_name = st.selectbox("Choose a table", tables)
             if table_name:
-                file_path = fetch_data_from_db(conn_str, table_name)
+                csv_path = fetch_data_from_db(conn_str, table_name)
                 st.success(f"Fetched **{table_name}** as CSV ✅")
 
 # ──────────────────────────────────────────────────────────────
-# ❷  OPTIONAL IMAGE UPLOAD + PREVIEW
+#  2.  OPTIONAL IMAGE UPLOAD & PREVIEW
 # ──────────────────────────────────────────────────────────────
 st.markdown("---")
 st.subheader("📷 Optional: Upload an Image for Preview / Future Analysis")
 
-uploaded_image = st.file_uploader(
-    "Upload an image (PNG or JPG)", type=["png", "jpg", "jpeg"], key="img"
+img_file = st.file_uploader(
+    "Upload an image (PNG/JPG)", type=["png", "jpg", "jpeg"], key="img"
 )
 
-if uploaded_image is not None:
-    image_path = os.path.join("data", uploaded_image.name)
-    with open(image_path, "wb") as f:
-        f.write(uploaded_image.read())
-    st.image(image_path, caption="Uploaded Image", use_column_width=True)
-
-    # 👉 If you later want to pass the image to an agent, you can add code here
-    # e.g. result = image_analysis_agent.run(input=image_path)
+if img_file:
+    img_path = os.path.join(TEMP_DIR, img_file.name)
+    with open(img_path, "wb") as f:
+        f.write(img_file.read())
+    st.image(img_path, caption="Uploaded Image", use_column_width=True)
+    # 👉  Future hook: pass `img_path` to a vision agent here.
 
 # ──────────────────────────────────────────────────────────────
-# ❸  RUN BUSINESS ANALYTICS PIPELINE
+#  3.  RUN ANALYTICS PIPELINE
 # ──────────────────────────────────────────────────────────────
-if file_path:
+if csv_path:
     st.markdown("---")
     st.info("Running analytics pipeline… ⏳")
-    result = analytics_coordinator.run(input=file_path)
+
+    report = analytics_coordinator.run(input=csv_path)
 
     st.subheader("📝 Analysis & Strategy Report")
-    st.text(result)
+    st.text(report)
 
-    # Display generated charts if they exist
-    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)
+    # Show charts generated by tool functions if present
+    for plot_name, caption in [
+        ("sales_plot.png", "Sales Trend"),
+        ("forecast_plot.png", "Forecast Chart"),
+    ]:
+        plot_path = os.path.join(TEMP_DIR, plot_name)
+        if os.path.exists(plot_path):
+            st.image(plot_path, caption=caption, use_column_width=True)