Upload app.py

app.py

@@ -1,4 +1,5 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from typing import List, Dict
 import time
@@ -33,6 +34,13 @@ def inference(question: str, tables: Dict[str, List[str]]) -> str:
     return result
 
 app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
 
 @app.get("/")
 def home():
@@ -43,7 +51,7 @@ def home():
         "description": "This api can be used to convert natural language to SQL given the human prompt, tables and the attributes."
     }
 
-@app.get("/generate")
+@app.get("/test-generate")
 def generate(text:str):
     start = time.time()
     res = inference("how many people with name jui and age less than 25", {
@@ -65,6 +73,33 @@ def generate(text:str):
         }
     }
 
+@app.post("/generate")
+def generate(request_body:Dict):
+    if 'text' not in request_body or 'tables' not in request_body:
+        raise HTTPException(status_code=400, detail="Missing 'text' or 'tables' in request body")
+    
+    prompt = request_body['text']
+    tables = request_body['tables']
+    
+    start = time.time()
+    res = inference(prompt, tables)
+    end = time.time()
+    total_time_taken = end - start
+    current_utc_datetime = datetime.datetime.now(datetime.timezone.utc)
+    current_date = datetime.date.today()
+    timezone_name = time.tzname[time.daylight]
+    print(res)
+    return {
+        "api_response": f"{res}",
+        "time_taken(s)": f"{total_time_taken}",
+        "request_details": {
+            "utc_datetime": f"{current_utc_datetime}",
+            "current_date": f"{current_date}",
+            "timezone_name": f"{timezone_name}"
+        }
+    }
+
+
 
 if __name__ == "__main__":
     uvicorn.run(app, host="127.0.0.1", port=8000)