Add sandbox

app.py

@@ -4,7 +4,8 @@ import requests
 import json
 import os
 from utils.google_genai_llm import get_response, generate_with_gemini
-from utils.utils import parse_json_codefences
+from utils.utils import parse_json_codefences, parse_python_codefences
+from utils.code_sandbox import code_eval
 from prompts.requirements_gathering import requirements_gathering_system_prompt
 from prompts.planning import hf_query_gen_prompt, hf_context_gen_prompt
 from prompts.devstral_coding_prompt import devstral_code_gen_sys_prompt, devstral_code_gen_user_prompt
@@ -198,7 +199,7 @@ def process_user_input(message, history, uploaded_files, file_cache):
     )
     
     # Get AI response
-    ai_response = get_response(formatted_prompt)
+    ai_response = generate_with_gemini(formatted_prompt, purpose="REQUIREMENTS_GATHERING")
     
     return ai_response, file_cache
 
@@ -348,7 +349,21 @@ def generate_code_with_devstral(plan_text, history, file_cache):
             return "❌ **Error:** No response received from Devstral model."
     except Exception as e:
         return f"❌ **Error:** {str(e)}"
-
+def execute_code(code_output):
+    try:
+        code = parse_python_codefences(code_output)
+        print(code)
+        result = code_eval(code)
+        if isinstance(result, dict):
+            result_str = json.dumps(result, indent=4)
+        elif isinstance(result, list):
+            result_str = '\n'.join(str(x) for x in result)
+        else:
+            result_str = str(result)
+        return result_str
+    except Exception as e:
+        return f"❌ **Error:** {str(e)}"
+        
 # Custom CSS for a sleek design
 custom_css = """
 .gradio-container {
@@ -447,23 +462,20 @@ with gr.Blocks(css=custom_css, title="Data Science Requirements Gathering Agent"
                 elem_classes=["chat-container"]
             )
 
-            plan_output = gr.Textbox(
+            plan_output = gr.Markdown(
                             label="Generated Plan",
-                            interactive=False,
                             visible=True,
-                            lines=10,
-                            max_lines=20
+                            max_height=150,
                         )
 
-            code_output = gr.Textbox(
+            code_output = gr.Markdown(
                             label="Generated Code",
-                            interactive=False,
-                            visible=True,
-                            lines=15,
-                            max_lines=30,
-                            placeholder="Generated Python code will appear here..."
+                            visible=True,max_height=150,
+                        )
+            execution_output = gr.Markdown(
+                            label="Execution Output",
+                            visible=True,max_height=150,
                         )
-            
             with gr.Row():
                 with gr.Column(scale=4):
                     msg = gr.Textbox(
@@ -520,6 +532,7 @@ with gr.Blocks(css=custom_css, title="Data Science Requirements Gathering Agent"
             with gr.Column():
                 plan_btn = gr.Button("Generate Plan 📋", variant="secondary", elem_classes=["btn-secondary"], size="lg")
                 code_btn = gr.Button("Generate Code 💻", variant="secondary", elem_classes=["btn-secondary"], size="lg")
+                execute_code_btn = gr.Button("Execute Code 🚀", variant="primary", elem_classes=["btn-primary"], size="lg")
     
     # State for conversation history and file cache
     chat_history = gr.State([])
@@ -561,7 +574,11 @@ with gr.Blocks(css=custom_css, title="Data Science Requirements Gathering Agent"
         inputs=[plan_output, chat_history, file_cache],
         outputs=[code_output]
     )
-    
+    execute_code_btn.click(
+        execute_code,
+        inputs=[code_output],
+        outputs=[execution_output]
+    )
     file_upload.change(
         lambda files: files,
         inputs=[file_upload],

prompts/devstral_coding_prompt.py

@@ -12,4 +12,6 @@ devstral_code_gen_user_prompt ="""
 
 **Context for Plan Steps**
 {context}
+
+Just return the full execution code block in a python codefence as shown below without any explanation or suffix or prefix text.
 """

utils/utils.py

@@ -21,3 +21,18 @@ def parse_json_codefences(text: str) -> List[Any]:
         except json.JSONDecodeError as e:
             print(f"⚠️ Failed to parse JSON block:\n{json_str}\nError: {e}")
     return results
+
+def parse_python_codefences(text: str) -> List[str]:
+    """
+    Extracts all ```python ... ``` code blocks from `text`
+    """
+    # Regex to match ```python ... ``` fences (non-greedy, multiline)
+    pattern = re.compile(r"```python\s+(.*?)\s+```", re.DOTALL | re.IGNORECASE)
+    results = ""
+    
+    for match in pattern.finditer(text):
+        python_str = match.group(1).strip()
+        results = python_str
+        
+    return results
+