Update app.py

app.py

@@ -6,13 +6,13 @@ import json
 import re
 
 # — Page config
-st.set_page_config(page_title="App with CSV Upload & AI Agent", layout="wide")
+st.set_page_config(page_title="CSV-Backed AI Agent", layout="wide")
 
-# — Title & image
-st.title("App with Form and CSV-Backed AI Agent")
+# — Title & (optional) image
+st.title("CSV-Backed AI Agent")
 st.image("./nadi-lok-image.png")
 
-# — Sidebar: CSV upload (from Code 1)
+# — Sidebar: CSV upload & preview (identical to your working code)
 st.sidebar.header("Upload CSV File")
 uploaded_file = st.sidebar.file_uploader("Choose a CSV file", type="csv")
 
@@ -22,6 +22,7 @@ if uploaded_file:
         st.sidebar.success("File uploaded successfully!")
         st.sidebar.write("Preview of the uploaded file:")
         st.sidebar.dataframe(df.head())
+        # Convert DataFrame back to CSV text to feed into the agent
         csv_text = df.to_csv(index=False)
     except Exception as e:
         st.sidebar.error(f"Error reading file: {e}")
@@ -31,53 +32,31 @@ else:
     df = None
     csv_text = None
 
-# — Main app: user form (from Code 1)
-with st.form("user_form"):
-    st.header("Welcome to Streamlit Form!")
-    user_input = st.text_input("Enter your name:")
-    age = st.slider("Select your age:", 0, 100, 25)
-    submit_button = st.form_submit_button("Submit")
-
-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.")
-
-# — 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
+# — Show basic info about the loaded CSV
 if df is not None:
     st.markdown(f"**Loaded CSV:** {df.shape[0]} rows × {df.shape[1]} columns")
 
-# Prompt input
+# — 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
+# — Run the AI Agent
 if st.button("Run Agent"):
+    # 1️⃣ Validation
     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
+        # 2️⃣ Build the "memory + prompt" 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)."
+                "Return your answer strictly as JSON (no additional explanation)."
             )
         }
         csv_msg = {
@@ -86,7 +65,12 @@ if st.button("Run Agent"):
         }
         user_msg = {"role": "user", "content": prompt}
 
-        # Call OpenAI API
+        # 3️⃣ Call OpenAI
+        api_key = os.getenv("OPENAI_API_KEY")
+        if not api_key:
+            st.error("❌ OPENAI_API_KEY not set in Secrets.")
+            st.stop()
+
         response = requests.post(
             "https://api.openai.com/v1/chat/completions",
             headers={
@@ -102,24 +86,27 @@ if st.button("Run Agent"):
             timeout=60,
         )
 
+        # 4️⃣ Handle API response
         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
+
+            # Strip any ``` fences and pull out the JSON object
             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]
+                # remove trailing commas
                 frag = re.sub(r',\s*([}\]])', r'\1', frag)
                 try:
                     parsed = json.loads(frag)
                 except json.JSONDecodeError:
                     parsed = None
 
-            # Display result
+            # 5️⃣ Display
             if parsed is not None:
                 st.subheader("✅ JSON Output")
                 st.json(parsed)