Update main.py

main.py

@@ -7,12 +7,12 @@ import pandas as pd
 from io import StringIO
 import os
 
-from pandasai import SmartDataframe
-import pandas as pd
-from pandasai.llm import OpenAI
-
+from llmlingua import PromptCompressor
+llm_lingua = PromptCompressor(
+    model_name="microsoft/llmlingua-2-bert-base-multilingual-cased-meetingbank",
+    use_llmlingua2=True, # Whether to use llmlingua-2
+)
 
-secret = os.environ["key"]
 
 app = FastAPI()
 app.add_middleware(
@@ -23,14 +23,9 @@ app.add_middleware(
     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)
+@app.post("/get_compressed_text")
+async def get_compressed_text(original_propmt):
+        compressed_prompt = llm_lingua.compress_prompt(original_propmt, instruction="", question="")
+        print("compressed")
+        return compressed_prompt
+