Update main.py

main.py

@@ -1,10 +1,13 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import os
+import gradio as gr
+import requests
 
+# 创建FastAPI应用
 app = FastAPI()
 
-# 定义请求体模型
+# 定义POST请求的JSON格式
 class RequestData(BaseModel):
     content: str
 
@@ -12,7 +15,6 @@ class RequestData(BaseModel):
 @app.post("/create_file")
 async def create_file(data: RequestData):
     try:
-        # 打开并写入文件
         with open("test.txt", "w") as file:
             file.write(data.content)
         return {"message": "File created successfully."}
@@ -23,13 +25,44 @@ async def create_file(data: RequestData):
 @app.get("/read_file")
 async def read_file():
     try:
-        # 检查文件是否存在
         if not os.path.exists("test.txt"):
             raise HTTPException(status_code=404, detail="File not found.")
         
-        # 读取文件内容
         with open("test.txt", "r") as file:
             content = file.read()
         return {"content": content}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+
+# 使用Gradio创建UI界面
+def api_request(url: str, request_type: str, json_data: str = None):
+    try:
+        if request_type == "POST":
+            # 如果是POST请求，解析JSON数据并发送
+            data = eval(json_data) if json_data else {}
+            response = requests.post(url, json=data)
+        else:
+            # 如果是GET请求，直接发送GET请求
+            response = requests.get(url)
+        
+        # 返回响应内容
+        return response.json() if response.ok else f"Error: {response.status_code} - {response.text}"
+    
+    except Exception as e:
+        return str(e)
+
+# 创建Gradio界面
+iface = gr.Interface(
+    fn=api_request, 
+    inputs=[
+        gr.Textbox(label="URL", placeholder="Enter the URL to send request to"),
+        gr.Radio(["GET", "POST"], label="Request Type"),
+        gr.Textbox(label="POST JSON Data (only for POST)", placeholder="Enter JSON data, e.g. {'content': 'Hello'}", visible=False)
+    ],
+    outputs="json",
+    live=True
+)
+
+# 启动Gradio界面
+iface.launch()
+