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)