Commit atualizado com o GET das imagens do projeto

backend/api.py

@@ -1,8 +1,99 @@
+from fastapi import FastAPI, BackgroundTasks
+from fastapi.responses import StreamingResponse
 import sys
 import os
+from datetime import datetime
+from io import BytesIO
+import zipfile
 
-parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
+
+parent_dir = os.path.abspath(os.path.join(os.path.dirname(_file_), os.pardir))
 sys.path.append(parent_dir)
 sys.path.append(os.path.join(parent_dir, 'models'))
 
+from status_return import *
 from models.generate_images import Generator
+
+#adding cors headers
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI
+
+app = FastAPI()
+g = Generator()
+
+database = g.get_database()
+
+# adding cors urls
+origins = [
+ 'http://localhost:3000'
+
+]
+
+# add middleware
+app.add_middleware(
+ CORSMiddleware,
+ allow_origins= origins, 
+ allow_credentials=True,
+ allow_methods=["*"], 
+ allow_headers=["*"]
+)
+
+
+@app.get('/create/{prompt}')
+async def create_infinite_zoom(prompt: str, background_tasks: BackgroundTasks):
+
+ if g.is_running():
+ return RUNNING
+
+ try:
+ prompt_gpt = await g.gpt_prompt_create(prompt)
+ now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+ project_id = database.insert_project(prompt, None, now, prompt_gpt)
+ background_tasks.add_task(g.generate_images, prompt_gpt, project_id)
+
+ return STARTED
+
+ except Exception as e: 
+ print(e)
+ return ERROR
+
+#! using this rout to save the images in a path on aria2
+@app.get('/savepath/{project_id}')
+async def save_image(project_id: int):
+ if not g.is_running():
+ g.read_image_from_db(project_id)
+ return 200
+ else:
+ return 400
+
+# Rota para buscar todas as imagens de um projeto com o parâmetro id_do_projeto
+@app.get('/get_images/{id_project}')
+async def get_images(id_project: int):
+
+ # Busca objetos de imagem diretamente do banco de dados
+ images = database.get_images_from_project(id_project)
+ if not images:
+ raise HTTPException(status_code=404, detail="No images found for this project")
+
+ # Object BytesIO -> ZIP em memória
+ memory_zip = BytesIO()
+
+ # Cria um novo arquivo ZIP no objeto BytesIO
+ with zipfile.ZipFile(memory_zip, 'w', zipfile.ZIP_DEFLATED) as zf:
+ for image_record in images:
+ # imagens estão em bytes.
+ img = Image.open(BytesIO(image_record.image_data))
+
+ # Converte para PNG e salva em um BytesIO temporário
+ img_bytes = BytesIO()
+ img.save(img_bytes, format='PNG')
+ img_bytes.seek(0)
+
+ # Adiciona ao ZIP
+ zf.writestr(f"{image_record.image_name}.png", img_bytes.getvalue())
+
+ # Apontar o ponteiro para o início do arquivo para leitura
+ memory_zip.seek(0)
+
+ # Retorna o arquivo ZIP como uma resposta de streaming
+ return StreamingResponse(memory_zip, media_type="application/zip")