Update app.py

app.py

@@ -1,6 +1,4 @@
-import os
 import subprocess
-import tempfile
 import sys
 from flask import Flask, request, jsonify
 from flask_cors import CORS
@@ -10,7 +8,7 @@ CORS(app)
 
 @app.route('/')
 def home():
-    return jsonify({"status": "ok", "message": "API Running. POST to /api/run-python or /api/run-sql"})
+    return jsonify({"status": "ok", "message": "API Running (Gunicorn Enabled)"})
 
 def run_command(command, input_text=None, timeout=15):
     """Helper to run shell commands safely"""
@@ -22,10 +20,8 @@ def run_command(command, input_text=None, timeout=15):
             text=True,
             timeout=timeout
         )
-        # Return stdout if success, stderr if fail (or both)
         output = run_process.stdout
         if run_process.stderr:
-             # Append stderr to output so user sees errors
              output += "\n--- Error Log ---\n" + run_process.stderr
         return output
     except subprocess.TimeoutExpired:
@@ -36,24 +32,15 @@ def run_command(command, input_text=None, timeout=15):
 @app.route('/api/run-python', methods=['POST'])
 def run_python():
     code = request.json.get('code', '')
-    user_input = request.json.get('input', '') # Optional stdin input
+    user_input = request.json.get('input', '')
     
     if not code: return jsonify({'output': 'No code provided'}), 400
 
-    # Write code to temp file
-    temp_path = None
-    try:
-        with tempfile.NamedTemporaryFile(suffix=".py", delete=False, mode='w', encoding='utf-8') as f:
-            f.write(code)
-            temp_path = f.name
-        
-        # Run python using the current environment's python executable
-        output = run_command([sys.executable, temp_path], input_text=user_input)
-        return jsonify({'output': output})
-        
-    finally:
-        if temp_path and os.path.exists(temp_path):
-            os.remove(temp_path)
+    # Run python using -c to avoid temp files
+    # We pass the code as the argument to -c
+    cmd = [sys.executable, "-c", code]
+    output = run_command(cmd, input_text=user_input)
+    return jsonify({'output': output})
 
 @app.route('/api/run-sql', methods=['POST'])
 def run_sql():
@@ -63,43 +50,31 @@ def run_sql():
 
     if not sql_code: return jsonify({'output': 'No SQL code provided'}), 400
 
-    # Write SQL to temp file (safer for some CLI tools)
-    temp_sql_path = None
-    try:
-        with tempfile.NamedTemporaryFile(suffix=".sql", delete=False, mode='w', encoding='utf-8') as f:
-            f.write(sql_code)
-            temp_sql_path = f.name
-
-        output = ""
-        
-        if db_type == 'sqlite':
-            # Run sqlite3 against an in-memory DB
-            # We pass the SQL content via stdin
-            cmd = ["sqlite3", ":memory:"]
-            output = run_command(cmd, input_text=sql_code)
-
-        elif db_type in ['postgres', 'postgresql']:
-            # Run psql using the 'playground' user defined in start.sh
-            # -f runs the file provided
-            cmd = ["psql", "-h", "localhost", "-U", "playground", "-d", "playground", "-f", temp_sql_path]
-            output = run_command(cmd)
+    output = ""
+    
+    if db_type == 'sqlite':
+        # SQL via Stdin -> SQLite
+        cmd = ["sqlite3", ":memory:"]
+        output = run_command(cmd, input_text=sql_code)
 
-        elif db_type == 'mysql':
-            # Run mysql using the 'playground' user defined in start.sh
-            # -t forces ASCII table output even in non-interactive mode
-            cmd = ["mysql", "-h", "localhost", "-u", "playground", "playground", "-t"]
-            # Pass SQL via stdin
-            output = run_command(cmd, input_text=sql_code)
+    elif db_type in ['postgres', 'postgresql']:
+        # SQL via Stdin -> Psql
+        # We do not use -f anymore. Psql reads from stdin by default.
+        cmd = ["psql", "-h", "localhost", "-U", "playground", "-d", "playground"]
+        output = run_command(cmd, input_text=sql_code)
 
-        else:
-            return jsonify({'output': 'Unsupported database type. Use sqlite, postgres, or mysql.'}), 400
+    elif db_type == 'mysql':
+        # SQL via Stdin -> MySQL
+        # -t forces table output
+        cmd = ["mysql", "-h", "localhost", "-u", "playground", "playground", "-t"]
+        output = run_command(cmd, input_text=sql_code)
 
-        return jsonify({'output': output})
+    else:
+        return jsonify({'output': 'Unsupported database type.'}), 400
 
-    finally:
-        if temp_sql_path and os.path.exists(temp_sql_path):
-            os.remove(temp_sql_path)
+    return jsonify({'output': output})
 
+# Note: app.run() is no longer needed here as Gunicorn handles it, 
+# but we keep it for local debugging if you run python app.py directly.
 if __name__ == '__main__':
-    # Hugging Face expects port 7860
     app.run(host="0.0.0.0", port=7860)