Upload 2 files

main.py

@@ -0,0 +1,55 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from typing import Dict, List
+import uuid
+
+app = FastAPI()
+
+# Estructura de datos en memoria
+salas: Dict[str, Dict] = {}
+
+# Modelos de datos
+class Mensaje(BaseModel):
+    usuario_id: str
+    texto: str
+
+class Jugador(BaseModel):
+    nombre: str
+
+@app.post("/sala")
+def crear_sala():
+    while True:
+        nueva_id = str(uuid.uuid4())[:8]
+        if nueva_id not in salas:
+            salas[nueva_id] = {"mensajes": [], "jugadores": {}}
+            return {"sala_id": nueva_id}
+
+@app.post("/sala/{sala_id}/registrar")
+def registrar_usuario(sala_id: str, jugador: Jugador):
+    if sala_id not in salas:
+        raise HTTPException(status_code=404, detail="Sala no encontrada")
+    usuario_id = str(uuid.uuid4())
+    salas[sala_id]["jugadores"][usuario_id] = jugador.nombre
+    return {"usuario_id": usuario_id, "nombre": jugador.nombre}
+
+@app.post("/sala/{sala_id}/mensaje")
+def enviar_mensaje(sala_id: str, mensaje: Mensaje):
+    if sala_id not in salas:
+        raise HTTPException(status_code=404, detail="Sala no encontrada")
+    if mensaje.usuario_id not in salas[sala_id]["jugadores"]:
+        raise HTTPException(status_code=403, detail="Usuario no registrado en la sala")
+    nombre = salas[sala_id]["jugadores"][mensaje.usuario_id]
+    salas[sala_id]["mensajes"].append({"usuario": nombre, "texto": mensaje.texto})
+    return {"ok": True}
+
+@app.get("/sala/{sala_id}/mensajes")
+def obtener_mensajes(sala_id: str):
+    if sala_id not in salas:
+        raise HTTPException(status_code=404, detail="Sala no encontrada")
+    return salas[sala_id]["mensajes"]
+
+@app.get("/sala/{sala_id}/jugadores")
+def obtener_jugadores(sala_id: str):
+    if sala_id not in salas:
+        raise HTTPException(status_code=404, detail="Sala no encontrada")
+    return list(salas[sala_id]["jugadores"].values())

requirements.txt

@@ -0,0 +1,7 @@
+fastapi
+uvicorn
+transformers
+torch
+protobuf==3.20.*
+accelerate>=0.26.0
+bitsandbytes