File size: 1,118 Bytes
d92c861 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
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)
|