|
from fastapi import FastAPI, Request, Header
|
|
from gradio_client import Client
|
|
from datetime import datetime
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
@app.api_route("/", methods=["GET", "POST"])
|
|
async def ping():
|
|
return {"pong": datetime.now().isoformat()}
|
|
|
|
|
|
@app.get("/private-policy")
|
|
async def ping():
|
|
return "We dont store your data. All data returned is public!"
|
|
|
|
@app.api_route("/gapi", methods=["GET", "POST"])
|
|
async def InvokeGradioApi(request: Request, x_hf_token: str = Header(default=None)):
|
|
if request.method == "POST":
|
|
data = await request.json()
|
|
else:
|
|
data = dict(request.query_params)
|
|
|
|
|
|
space_name = data.pop("space", None)
|
|
auth_key = data.pop("auth", None) or x_hf_token
|
|
|
|
if not space_name:
|
|
return {"error": "Missing required parameter: space"}
|
|
|
|
print("Creating client", space_name);
|
|
client = Client(space_name, hf_token=auth_key if auth_key else None)
|
|
|
|
print("Data:")
|
|
print(data);
|
|
|
|
try:
|
|
result = client.predict(**data)
|
|
return result
|
|
except Exception as e:
|
|
return {"error": str(e)}
|
|
|
|
|
|
|
|
@app.get("/SearchScripts")
|
|
async def SearchScript(request: Request, x_hf_token: str = Header(default=None)):
|
|
data = dict(request.query_params)
|
|
|
|
auth_key = data.pop("auth", None) or x_hf_token
|
|
client = Client('rrg92/sqlserver-lib-assistant', hf_token=auth_key if auth_key else None)
|
|
|
|
search_text = data['text'];
|
|
|
|
print("search", search_text);
|
|
|
|
try:
|
|
result = client.predict(message=search_text, api_name="/SearchFiles")
|
|
return result
|
|
except Exception as e:
|
|
print(e);
|
|
return {"error": "Some error happened. Admin must check logs"} |
