Update app.py

app.py

@@ -1,29 +1,52 @@
 import gradio as gr
 import requests
+from fastapi import FastAPI, Request
+from fastapi.responses import Response
 
-def proxy_request(path=""):
+app = FastAPI()
+
+@app.get("/{full_path:path}")
+@app.post("/{full_path:path}")
+async def proxy(full_path: str, request: Request):
     try:
-        if path.startswith("/api/"):
-            # 如果路径以 /api/ 开头，去掉 /api/ 后转发
-            actual_path = path[5:]  # 跳过 "/api/"
-            url = f"http://beibeioo.top/{actual_path}"
+        # 构建目标 URL
+        if full_path.startswith("api/"):
+            # 如果路径以 api/ 开头，去掉 api/ 后转发
+            actual_path = full_path[4:]  # 跳过 "api/"
+            target_url = f"http://beibeioo.top/{actual_path}"
         else:
             # 否则直接访问根路径
-            url = "http://beibeioo.top"
-            
-        response = requests.get(url)
-        return response.text
+            target_url = "http://beibeioo.top"
+
+        # 转发请求
+        response = requests.request(
+            method=request.method,
+            url=target_url,
+            headers={
+                'User-Agent': 'HuggingFace-Space-Proxy',
+                'Accept': '*/*'
+            }
+        )
+
+        # 返回响应
+        return Response(
+            content=response.content,
+            status_code=response.status_code,
+            headers={
+                'Content-Type': response.headers.get('Content-Type', 'application/json'),
+                'Access-Control-Allow-Origin': '*'
+            }
+        )
     except Exception as e:
-        return f"Error: {str(e)}"
+        return {"error": str(e)}
 
-# 创建 Gradio 接口
+# 为了保持 Gradio UI 可见，添加一个简单的界面
 demo = gr.Interface(
-    fn=proxy_request,
-    inputs=gr.Textbox(label="Path (leave empty for root, or use /api/v1 for API)"),
-    outputs=gr.Textbox(),
+    fn=lambda x: x,
+    inputs=gr.Textbox(visible=False),
+    outputs=gr.Textbox(visible=False),
     title="API Proxy",
     description="Access beibeioo.top through this proxy"
 )
 
-if __name__ == "__main__":
-    demo.launch()
+app = gr.mount_gradio_app(app, demo, path="/ui")