Update app.py

app.py

@@ -1,9 +1,25 @@
+import logging
 import gradio as gr
+from fastapi import Request
+from fastapi.responses import JSONResponse
 import sys
 sys.path.append("scripts/")
 from lib import init, refresh_character_thumb_image, get_prompt_manager
 from lib import TITLE
 
+logging.getLogger("gradio").setLevel(logging.ERROR)
+logging.getLogger("uvicorn.error").setLevel(logging.ERROR)
+
+async def security_middleware(request: Request, call_next):    
+    if request.method == "POST" and "/api/" in request.url.path:
+        return JSONResponse(status_code=403, content={"error": "Forbidden"})
+    
+    try:
+        return await call_next(request)
+    except:
+        return JSONResponse(status_code=400, content={"error": "Bad Request"})
+
+
 if __name__ == '__main__':
     character_list, character_list_cn, LANG, JAVA_SCRIPT, CSS_SCRIPT = init()
     
@@ -59,4 +75,5 @@ if __name__ == '__main__':
         # Use custom_prompt, the stupid js console will always report "api_info.ts:423  Too many arguments provided for the endpoint."
         dummy_textbox.change(fn=get_prompt_manager().update_suggestions_js, inputs=[dummy_textbox], outputs=[dummy_dropdown])
         
-    ui.launch(show_api=False)
+    ui.app.middleware("http")(security_middleware)
+    ui.launch(show_api=False, share=False)