feature(#86): implement debug ui for sendNotification function

Brain/src/gradio_debug.py

@@ -0,0 +1,41 @@
+"""gradio implementaion"""
+import gradio as gr
+import requests
+import json
+
+DEFAULT_UUID = "c40a09075d11940f"
+
+
+def gradio_send_notification(uuid: str, query: str):
+    url = "http://localhost:7860/sendNotification"
+    data = {
+        "confs": {
+            "openai_key": "",
+            "pinecone_key": "",
+            "pinecone_env": "",
+            "firebase_key": "",
+            "settings": {"temperature": 0.6},
+            "token": "eSyP3i7ITZuq8hWn2qutTl:APA91bH1FtWkaTSJwuX4WKWAl3Q-ZFyrOw4UtMP4IfwuvNrHOThH7EvEGIhtguilLRyQNlLiXatEN0xntHAc8bbKobSGjge3wxIHlspbIWY_855CzONqaVdl3y3zOmgKZNnuhYi4gwbh",
+            "uuid": uuid,
+        },
+        "message": query,
+    }
+
+    response = requests.post(
+        url, data=json.dumps(data), headers={"Content-Type": "application/json"}
+    )
+    response = response.json()
+    return json.dumps(response["result"]), response["status_code"]
+
+
+debug_send_notification = gr.Interface(
+    fn=gradio_send_notification,
+    inputs=[
+        gr.Textbox(
+            label="UUID", placeholder="Please input your UUID.", value=DEFAULT_UUID
+        ),
+        gr.Textbox(label="query", placeholder="Please input your prompt."),
+    ],
+    outputs=[gr.Textbox(label="Result"), gr.Number(label="Status Code")],
+    allow_flagging="never",
+)

app.py

@@ -3,10 +3,14 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
 
+from Brain.src.gradio_debug import debug_send_notification
 from Brain.src.router.browser_router import construct_blueprint_browser_api
 from Brain.src.router.train_router import construct_blueprint_train_api
 
 from Brain.src.router.api import construct_blueprint_api
+import gradio as gr
+import random
+import time
 
 app = FastAPI()
 # Set up CORS middleware
@@ -26,6 +30,13 @@ app.include_router(
 
 app.include_router(construct_blueprint_train_api(), prefix="/train", tags=["ai_train"])
 
+# gradio
+
+CUSTOM_PATH = "/gradio"
+
+app = gr.mount_gradio_app(
+    app, debug_send_notification, path=f"{CUSTOM_PATH}/sendNotification"
+)
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -72,4 +72,6 @@ user-agents==2.2.0
 faiss-cpu==1.7.4
 google-search-results==2.4.2
 tiktoken==0.4.0
-tiktoken==0.4.0
+tiktoken==0.4.0
+gradio-client==0.2.7
+gradio==3.35.2