Update app.py

app.py

@@ -1,40 +1,37 @@
-from flask import Flask, request, jsonify
-import sys
+from flask import Flask, request, jsonify, render_template
 import io
-import contextlib
+import sys
 
 app = Flask(__name__)
 
-# Function to execute Python code safely
-def execute_python_code(code):
-    # Capturing the output of Python execution
+@app.route("/")
+def index():
+    # Render the terminal-like interface
+    return render_template("index.html")
+
+@app.route("/execute", methods=["POST"])
+def execute_code():
+    # Get the Python code sent from the frontend
+    code = request.json.get("code", "")
+    
+    # Redirect standard output to capture execution output
     output = io.StringIO()
     sys.stdout = output
     sys.stderr = output
-
+    
     try:
-        # Executing the code
+        # Execute the provided code
         exec(code)
+        result = output.getvalue()
     except Exception as e:
-        output.write(f"Error: {e}")
-
-    # Return the captured output
-    return output.getvalue()
-
-@app.route('/')
-def index():
-    return app.send_static_file('index.html')
+        result = f"Error: {e}"
+    finally:
+        # Reset standard output
+        sys.stdout = sys.__stdout__
+        sys.stderr = sys.__stderr__
 
-@app.route('/execute', methods=['POST'])
-def execute():
-    data = request.get_json()
-    code = data.get('code', '')
-    
-    # Execute the Python code and capture the result
-    result = execute_python_code(code)
-    
-    # Send the result back to the frontend
-    return jsonify({'result': result})
+    return jsonify({"result": result})
 
-if __name__ == '__main__':
-    app.run(debug=True)
+if __name__ == "__main__":
+    # Run the app on host 0.0.0.0 and port 7860 for compatibility with Hugging Face Spaces
+    app.run(host="0.0.0.0", port=7860)