import time
from pydantic import BaseModel
import base64
from fastapi import FastAPI, APIRouter
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI()

origins = [
    "http://localhost:7860",  # Exemple d'origine (ajoutez vos origines ici)
    "https://example.com",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,  # Liste des origines autorisées
    allow_credentials=True,
    allow_methods=["*"],  # Autorise toutes les méthodes (GET, POST, etc.)
    allow_headers=["*"],  # Autorise tous les en-têtes
)

class ImageData(BaseModel):
    image: str

class ImagesData(BaseModel):
    idCard: str
    profileImage: str

@app.post("/uploadpdf")
async def upload_pdf(image_data: ImageData):
    # Extraire le contenu Base64
    header, encoded = image_data.image.split(',', 1)  # Séparer le header des données
    binary_data = base64.b64decode(encoded)  # Décoder les données

    time.sleep(20);

    return {"message": "Image reçue et sauvegardée"}

@app.post("/uploadids")
async def upload_ids(images_data: ImagesData):
    header, encoded = images_data.idCard.split(',', 1)  # Séparer le header des données
    id_card_binary_data = base64.b64decode(encoded)

    header, encoded = images_data.idCard.split(',', 1)  # Séparer le header des données
    profile_image_binary_data = base64.b64decode(encoded)

    time.sleep(20);

    return {"message": "Image reçue et sauvegardée"}