| 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"} |
