Update main.py

main.py

@@ -1,20 +1,35 @@
-from fastapi import FastAPI
-from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse
-
-from transformers import pipeline
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import os
 
 app = FastAPI()
 
-pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
-
-@app.get("/infer_t5")
-def t5(input):
-    output = pipe_flan(input)
-    return {"output": output[0]["generated_text"]}
+# 定义请求体模型
+class RequestData(BaseModel):
+    content: str
 
-app.mount("/", StaticFiles(directory="static", html=True), name="static")
+# POST请求：创建test.txt文件并写入内容
+@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."}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-@app.get("/")
-def index() -> FileResponse:
-    return FileResponse(path="/app/static/index.html", media_type="text/html")
+# GET请求：读取test.txt文件内容
+@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))