Update main.py

main.py

@@ -1,13 +1,46 @@
-from fastapi import FastAPI, Query, Request
-from fastapi.responses import JSONResponse, HTMLResponse
+from fastapi import FastAPI, Request, File, UploadFile
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+import os
 
 app = FastAPI()
 
+# Директория для временного хранения загруженных файлов
+UPLOAD_DIRECTORY = "uploads"
+os.makedirs(UPLOAD_DIRECTORY, exist_ok=True)
 
-@app.get("/infer_t5")
-def t5(input):
-    return {"output": "output"}
+# Подключаем статические файлы (например, CSS)
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
+# Подключаем шаблоны Jinja2
+templates = Jinja2Templates(directory="templates")
+
+
+# Главная страница с текстом "server is running"
 @app.get("/", response_class=HTMLResponse)
-async def streamlit_proxy(request: Request):
-    return 'Server is running'
+async def read_root():
+    return templates.TemplateResponse("index.html", {"request": request, "text": "server is running"})
+
+
+# Первый POST эндпоинт для приема текста
+@app.post("/update_text/")
+async def update_text(text: str):
+    return {"text": text}
+
+
+# Второй POST эндпоинт для приема файла и текстового поля prompt
+@app.post("/upload_file/")
+async def upload_file(prompt: str, file: UploadFile = File(...)):
+    # Сохраняем файл в директорию UPLOAD_DIRECTORY
+    file_location = os.path.join(UPLOAD_DIRECTORY, file.filename)
+    with open(file_location, "wb") as file_object:
+        file_object.write(file.file.read())
+
+    # Отправляем запрос на первый эндпоинт с текстом равным имени загруженного файла
+    response = await update_text(file.filename)
+
+    # Удаляем загруженный файл
+    os.remove(file_location)
+
+    return response