Update app.py

app.py

@@ -25,6 +25,9 @@ class Ticket(BaseModel):
 
 class Code(BaseModel):
     code: str
+    project_id: str
+    chain_id: str
+    session_id: str
 
 @app.get("/")
 def greet_json():
@@ -48,7 +51,7 @@ async def run_code(code: Code):
     exec(code.code)
     # img_buffer.seek(0)  # Reset buffer position
 
-    file_path = "graph.pdf"
+    file_path = f"graphs_{code.project_id}_{code.chain_id}_{code.session_id}.pdf"
 
     # if "plt.subplots(" in code.code:
     #     print("SUBPLOTS DETECTED")
@@ -76,4 +79,14 @@ async def run_code(code: Code):
     
     # return Response(content=img_buffer.getvalue(), media_type="image/png")
     # return FileResponse(file_path, media_type="image/png")
-    return FileResponse(file_path, media_type="application/pdf", filename="graph.pdf")
+    return FileResponse(file_path, media_type="application/pdf", filename="graph.pdf")
+
+
+
+@app.get("/chart/{project_id}/{chain_id}")
+async def get_chart(project_id: str, chain_id: str, session_id: str):
+    
+    # Return as application/pdf response
+    return Response(pdf_buffer.read(), media_type="application/pdf", headers={
+        "Content-Disposition": f'inline; filename="graphs_{project_id}_{chain_id}_{session_id}.pdf"'
+    })