Spaces:
Running
Running
| import os | |
| import uvicorn | |
| from fastapi import FastAPI | |
| from fastapi.middleware.cors import CORSMiddleware | |
| from openai import OpenAI | |
| from pydantic import BaseModel | |
| from typing import List | |
| # 1. Configuration des accès | |
| # Assurez-vous d'avoir défini ALBERT_API_KEY dans vos secrets Hugging Face | |
| ALBERT_API_KEY = os.getenv("ALBERT_API_KEY") | |
| ALBERT_BASE_URL = "https://albert.api.etalab.gouv.fr/v1" | |
| # Initialisation du client compatible Albert | |
| client = OpenAI( | |
| api_key=ALBERT_API_KEY, | |
| base_url=ALBERT_BASE_URL | |
| ) | |
| app = FastAPI() | |
| # 2. Configuration CORS pour autoriser votre frontend React | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=["*"], # En production, remplacez par l'URL de votre front | |
| allow_credentials=True, | |
| allow_methods=["*"], | |
| allow_headers=["*"], | |
| ) | |
| # Modèle de données pour valider les requêtes entrantes | |
| class Message(BaseModel): | |
| role: str | |
| content: str | |
| class ChatRequest(BaseModel): | |
| messages: List[Message] | |
| async def chat_albert(request: ChatRequest): | |
| try: | |
| # On utilise l'alias simplifié qui fonctionne sur tous les comptes Albert | |
| response = client.chat.completions.create( | |
| model="mistralai/Mistral-Small-3.2-24B-Instruct-2506", | |
| messages=[{"role": m.role, "content": m.content} for m in request.messages], | |
| temperature=0.7, | |
| max_tokens=1000 | |
| ) | |
| return {"content": response.choices[0].message.content} | |
| except Exception as e: | |
| # On imprime l'erreur exacte dans les logs de Hugging Face pour pouvoir la lire | |
| response = client.chat.completions.create(...) | |
| print(f"RÉPONSE IA : {response.choices[0].message.content}") # Ajoute ça ! | |
| return {"content": response.choices[0].message.content} | |
| if __name__ == "__main__": | |
| # Hugging Face utilise souvent le port 7860 | |
| uvicorn.run(app, host="0.0.0.0", port=7860) |