Update app.py

app.py

@@ -47,12 +47,14 @@ class S3DirectStream:
             config_file = hf_hub_download(repo_id=model_name, filename="config.json", token=HUGGINGFACE_HUB_TOKEN)
             tokenizer_file = hf_hub_download(repo_id=model_name, filename="tokenizer.json", token=HUGGINGFACE_HUB_TOKEN)
 
-            # Verificar si el archivo ya existe en S3
+            # Verificar si la carpeta y los archivos ya existen en S3
             if not await self.file_exists_in_s3(f"{model_name}/config.json"):
+                self.create_folder_if_not_exists(model_name)
                 with open(config_file, "rb") as file:
                     self.s3_client.put_object(Bucket=self.bucket_name, Key=f"{model_name}/config.json", Body=file)
 
             if not await self.file_exists_in_s3(f"{model_name}/tokenizer.json"):
+                self.create_folder_if_not_exists(model_name)
                 with open(tokenizer_file, "rb") as file:
                     self.s3_client.put_object(Bucket=self.bucket_name, Key=f"{model_name}/tokenizer.json", Body=file)
 
@@ -67,6 +69,15 @@ class S3DirectStream:
         except self.s3_client.exceptions.ClientError:
             return False
 
+    def create_folder_if_not_exists(self, model_name):
+        try:
+            # Las carpetas no existen como tal en S3, pero se pueden crear archivos vacíos para simular carpetas
+            # Crear un archivo vacío para simular la carpeta
+            self.s3_client.put_object(Bucket=self.bucket_name, Key=f"{model_name}/")
+        except Exception as e:
+            logging.error(f"Error al crear la carpeta en S3: {e}")
+            raise HTTPException(status_code=500, detail=f"Error al crear la carpeta en S3: {str(e)}")
+
     async def load_model_from_s3(self, model_name):
         try:
             model_name = model_name.replace("/", "-").lower()