Upgraded gradio interface

app.py

@@ -370,24 +370,55 @@ def execute_code(code_output):
 
         yield "**⏳ Executing code...** (Parsing code)"
         code = parse_python_codefences(code_output)
-        print(code)
         
         if not code or not code.strip():
             yield "❌ No Python code found to execute."
             return
         
         yield "**⏳ Executing code...** (Running in sandbox)"
-        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)
+        exec_result, build_logs = code_eval(code)
+
+        # Ensure exec_result is a dictionary
+        if not isinstance(exec_result, dict):
+            yield f"❌ **Error:** Unexpected execution result format.\\n\\n```\\n{str(exec_result)}\\n```"
+            return
+            
+        return_code = exec_result.get('returncode', -1)
+        stdout = exec_result.get('stdout', '')
+        stderr = exec_result.get('stderr', '')
+        error_msg = exec_result.get('error', 'Unknown error')
+
+        # Build the formatted Markdown output
+        formatted_output = ""
+        if return_code == 0:
+            formatted_output += "## ✅ Execution Successful\n"
+            if stdout:
+                formatted_output += f"**Output:**\n```text\n{stdout.strip()}\n```\n"
+            if stderr:
+                formatted_output += f"**Warnings (`stderr`):**\n```text\n{stderr.strip()}\n```\n"
         else:
-            result_str = str(result)
+            formatted_output += f"## ❌ Execution Failed (Exit Code: {return_code})\n"
+            formatted_output += f"**Error:** `{error_msg}`\n\n"
+            if stderr:
+                formatted_output += f"**Error Log (`stderr`):**\n```text\n{stderr.strip()}\n```\n"
+            if stdout:
+                formatted_output += f"**Output (`stdout`):**\n```text\n{stdout.strip()}\n```\n"
+
+        # Add build logs in a collapsible section
+        if build_logs:
+            formatted_output += f"""
+<details>
+<summary>Click to view build logs</summary>
+
+```
+{build_logs.strip()}
+```
+</details>
+"""
+        yield formatted_output
 
-        yield f"**✅ Execution Complete:**\n\n```\n{result_str}\n```"
     except Exception as e:
-        yield f"❌ **Error executing code:** {str(e)}"
+        yield f"❌ **Error running execution logic:** {str(e)}\n\n{traceback.format_exc()}"
         
 # Custom CSS for a sleek design
 custom_css = """

prompts/devstral_coding_prompt.py

@@ -15,5 +15,9 @@ devstral_code_gen_user_prompt ="""
 
 Just return the full execution code block in a python codefence as shown below without any explanation or suffix or prefix text.
 
-Ensure that the code is EXECUTABLE and does not contain any errors.
+Ensure that the code is EXECUTABLE and does not contain any errors. 
+
+Refrain from and DO NOT hallucinate/make up datasets or models. If you dont have any relevant dataset generate synthetic data and use that to build a model!
+
+If you get a dataset from the Hugging Face Hub, make sure to check the exact columns of the dataset to use it properly.
 """