Update app.py

app.py

@@ -5,6 +5,7 @@ import subprocess
 import os
 import io
 import contextlib
+import re
 
 # Setup OpenAI
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -13,9 +14,12 @@ st.set_page_config(page_title="🧠 Build Any Model", layout="centered")
 st.title("🤖 Dynamic Code Generator Chatbot")
 
 prompt = st.text_input("🧠 Tell me what you want to build:")
-
 uploaded_file = st.file_uploader("📂 Upload your dataset (CSV)", type=["csv"])
 
+def replace_dataset_paths(code: str) -> str:
+    """Replace all pd.read_csv(...) with in-memory file reference 'uploaded_file'."""
+    return re.sub(r"pd\.read_csv\(['\"](.*?)['\"]\)", "pd.read_csv(uploaded_file)", code)
+
 if st.button("🚀 Generate and Run"):
     if prompt and uploaded_file:
         df = pd.read_csv(uploaded_file)
@@ -35,14 +39,17 @@ if st.button("🚀 Generate and Run"):
         code = response.choices[0].message.content
         if "```python" in code:
             code = code.split("```python")[-1].split("```")[0].strip()
+
+        # 🔁 Replace any pd.read_csv(...) with the uploaded_file reference
+        code = replace_dataset_paths(code)
         
         st.code(code, language="python")
-        
-        # Execute the generated code
+
+        # 🔄 Inject uploaded_file as a runtime variable
         output_buffer = io.StringIO()
         try:
             with contextlib.redirect_stdout(output_buffer):
-                exec(code, {"df": df})
+                exec(code, {"uploaded_file": uploaded_file})
             st.success("✅ Code executed successfully.")
             st.text(output_buffer.getvalue())
         except Exception as e: