Update main.py

main.py

@@ -0,0 +1,36 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import HTMLResponse
+from fastapi.responses import StreamingResponse
+from fastapi.responses import FileResponse
+from fastapi.middleware.cors import CORSMiddleware
+import pandas as pd
+from io import StringIO
+import os
+
+from pandasai import SmartDataframe
+import pandas as pd
+from pandasai.llm import OpenAI
+
+
+secret = os.environ["key"]
+
+app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.post("/get_image_for_text")
+async def get_image_for_text(email,query,file: UploadFile = File(...)):
+        print(file.filename)
+        with open(email+".csv", "wb") as file_object:
+            file_object.write(file.file.read())
+        llm = OpenAI(api_token=secret,save_charts=True)
+        df = pd.read_csv(email+".csv") 
+        sdf = SmartDataframe(df, config={"llm": llm})
+        sdf.chat(query)
+        image_path = "exports/charts/temp_chart.png"  # Replace with your image's path
+        return FileResponse(image_path)