Update main.py

main.py

@@ -3,6 +3,7 @@ from fastapi.responses import HTMLResponse
 from fastapi.responses import StreamingResponse
 from fastapi.responses import FileResponse
 from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 from io import StringIO
 import os
 
@@ -23,9 +24,11 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+class Prompt(BaseModel:
+    original_prompt: str
 @app.post("/get_compressed_text")
-async def get_compressed_text(original_propmt):
-        compressed_prompt = llm_lingua.compress_prompt(original_propmt, instruction="", question="")
+async def get_compressed_text(prompt: Prompt):
+        compressed_prompt = llm_lingua.compress_prompt(prompt.original_propmt, instruction="", question="")
         print("compressed")
         return compressed_prompt