Update app.py

app.py

@@ -1,45 +1,128 @@
 import streamlit as st
 import pandas as pd
+import os
+import requests
+import json
+import re
 
-# Set the title of the app
-st.title("App with Form and File Upload")
+# — Page config
+st.set_page_config(page_title="App with CSV Upload & AI Agent", layout="wide")
 
-# Display an image
+# — Title & image
+st.title("App with Form and CSV-Backed AI Agent")
 st.image("./nadi-lok-image.png")
 
-# Sidebar for file upload
+# — Sidebar: CSV upload (from Code 1)
 st.sidebar.header("Upload CSV File")
 uploaded_file = st.sidebar.file_uploader("Choose a CSV file", type="csv")
 
-# Display the uploaded file's content if a file is uploaded
 if uploaded_file:
-    # Read the CSV file
     try:
-        data = pd.read_csv(uploaded_file)
+        df = pd.read_csv(uploaded_file)
         st.sidebar.success("File uploaded successfully!")
         st.sidebar.write("Preview of the uploaded file:")
-        st.sidebar.dataframe(data.head())  # Display the first few rows of the data
+        st.sidebar.dataframe(df.head())
+        csv_text = df.to_csv(index=False)
     except Exception as e:
         st.sidebar.error(f"Error reading file: {e}")
+        df = None
+        csv_text = None
+else:
+    df = None
+    csv_text = None
 
-# Main app - Create a form
+# — Main app: user form (from Code 1)
 with st.form("user_form"):
-    # Display a header
     st.header("Welcome to Streamlit Form!")
-
-    # Add a text input widget
     user_input = st.text_input("Enter your name:")
-
-    # Add a slider widget
     age = st.slider("Select your age:", 0, 100, 25)
-
-    # Add form buttons
     submit_button = st.form_submit_button("Submit")
 
-# Handle form submission
 if submit_button:
     if user_input:
         st.success(f"Hello, {user_input}!")
         st.info(f"You are {age} years old.")
     else:
-        st.error("Please enter your name.")
+        st.error("Please enter your name.")
+
+# — AI Agent section (from Code 2)
+st.subheader("🔎 CSV-Backed AI Agent")
+
+# Load OpenAI key
+api_key = os.getenv("OPENAI_API_KEY")
+if not api_key:
+    st.error("❌ Secret OPENAI_API_KEY not set! Add it under Settings → Secrets.")
+    st.stop()
+
+# Show loaded CSV info
+if df is not None:
+    st.markdown(f"**Loaded CSV:** {df.shape[0]} rows × {df.shape[1]} columns")
+
+# Prompt input
+prompt = st.text_area(
+    "Enter your prompt for the agent",
+    placeholder="e.g. Which products have price > 100?",
+    height=150
+)
+
+# Run the agent
+if st.button("Run Agent"):
+    if df is None:
+        st.error("Please upload a CSV file first.")
+    elif not prompt.strip():
+        st.error("Please enter a prompt.")
+    else:
+        # Build chat messages
+        system_msg = {
+            "role": "system",
+            "content": (
+                "You are an AI agent that reads the provided CSV data and answers the user's query. "
+                "Return your answer as JSON only (no extra text)."
+            )
+        }
+        csv_msg = {
+            "role": "system",
+            "content": f"CSV Data:\n{csv_text}\n<End of CSV>"
+        }
+        user_msg = {"role": "user", "content": prompt}
+
+        # Call OpenAI API
+        response = requests.post(
+            "https://api.openai.com/v1/chat/completions",
+            headers={
+                "Authorization": f"Bearer {api_key}",
+                "Content-Type": "application/json",
+            },
+            json={
+                "model": "gpt-3.5-turbo",
+                "messages": [system_msg, csv_msg, user_msg],
+                "temperature": 0,
+                "max_tokens": 1500,
+            },
+            timeout=60,
+        )
+
+        if response.status_code != 200:
+            st.error(f"API Error {response.status_code}: {response.text}")
+        else:
+            answer = response.json()["choices"][0]["message"]["content"]
+            # Strip code fences & extract JSON
+            txt = re.sub(r'```(?:json)?', '', answer).strip()
+            start = txt.find("{")
+            end = txt.rfind("}") + 1
+            parsed = None
+            if start >= 0 and end > 0:
+                frag = txt[start:end]
+                frag = re.sub(r',\s*([}\]])', r'\1', frag)
+                try:
+                    parsed = json.loads(frag)
+                except json.JSONDecodeError:
+                    parsed = None
+
+            # Display result
+            if parsed is not None:
+                st.subheader("✅ JSON Output")
+                st.json(parsed)
+            else:
+                st.subheader("🔍 Raw Output")
+                st.text(answer)