Update app.py

app.py

@@ -3,6 +3,11 @@ from pydantic import BaseModel
 
 import random
 
+import matplotlib.pyplot as plt
+import pandas as pd
+
+import io
+
 app = FastAPI()
 
 # Define the ticket schema using Pydantic
@@ -14,6 +19,10 @@ class Ticket(BaseModel):
     service_category: str
     difficulty: int  # Adjust type as needed (e.g., int or str)
 
+
+class Code(BaseModel):
+    code: str
+
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
@@ -28,3 +37,13 @@ async def create_ticket(ticket: Ticket):
         "message": "Ticket created successfully",
         "ticket": tick
     }
+
+
+@app.post("/run_code")
+async def run_code(code: Code):
+    img_buffer = io.BytesIO()
+    exec(code.code)
+    img_buffer.seek(0)  # Reset buffer position
+
+    # Return image as response
+    return Response(content=img_buffer.getvalue(), media_type="image/png")