Upload app.py

app.py

@@ -13,6 +13,8 @@ class QueryRequest(BaseModel):
 @app.get("/")
 def home():
     return {"message": "SQL Generation Server is running"}
+
+@app.post("/generate")
 def generate(request: QueryRequest):
     try:
         text = request.text
@@ -22,7 +24,6 @@ def generate(request: QueryRequest):
         generated_text = output[0]['generated_text']
         sql_query = generated_text.split("SQL query:")[-1].strip()
         
-
         # Basic validation to ensure it's a valid SQL query
         if not sql_query.lower().startswith(('select', 'show', 'describe', 'insert', 'update', 'delete')):
             raise ValueError("Generated text is not a valid SQL query")
@@ -44,8 +45,6 @@ def generate(request: QueryRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-
-
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)@app.post("/generate")
+    uvicorn.run(app, host="0.0.0.0", port=7860)