Update app.py

app.py

@@ -1,15 +1,26 @@
-import pygame
-import gradio as gr
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+import sys
+import io
+import traceback
 
-def run_game(code):
-    pygame.init()
-    screen = pygame.display.set_mode((400, 300))
-    pygame.display.set_caption("Headless Game")
-    screen.fill((0, 128, 255))
-    pygame.draw.circle(screen, (255, 0, 0), (200, 150), 75)
-    pygame.image.save(screen, "output.png")
-    pygame.quit()
-    return "output.png"
+app = FastAPI()
 
-iface = gr.Interface(fn=run_game, inputs="text", outputs="image")
-iface.launch()
+@app.post("/execute")
+async def execute_code(request: Request):
+    data = await request.json()
+    code = data.get("code", "")
+    output = io.StringIO()
+    sys_stdout = sys.stdout
+    sys.stderr = sys.stderr
+    sys.stdout = output
+    sys.stderr = output
+    try:
+        exec(code, {})
+        result = output.getvalue()
+    except Exception:
+        result = output.getvalue() + "\n" + traceback.format_exc()
+    finally:
+        sys.stdout = sys_stdout
+        sys.stderr = sys.stderr
+    return JSONResponse({"result": resul