Update app.py

app.py

@@ -10,14 +10,31 @@ import requests
 from io import BytesIO
 
 
+
+import gradio as gr
+import os
+from huggingface_hub import HfFileSystem, HfApi
+
 # Récupérer token depuis les variables d'environnement
 hf_token = os.getenv("MisterAI_bigscience_bloom_560m")
 
-# Configurer le token pour l'utilisation avec Hugging Face
-if hf_token:
-    HfFolder.save_token(hf_token)
-else:
-    raise ValueError("Le token Hugging Face n'est pas configuré. Assurez-vous qu'il est défini dans les variables d'environnement.")
+# Fenêtre d'authentification
+with gr.Blocks() as auth_demo:
+    hf_token_input = gr.Textbox(label="Hugging Face Token")
+    auth_button = gr.Button("Authentifier")
+
+def authenticate(hf_token):
+    # Configurer le token pour l'utilisation avec Hugging Face
+    if hf_token:
+        HfFolder.save_token(hf_token)
+    else:
+        raise ValueError("Le token Hugging Face n'est pas configuré. Assurez-vous qu'il est défini dans les variables d'environnement.")
+
+auth_button.click(
+    authenticate,
+    inputs=hf_token_input,
+    outputs=gr.Textbox(value="Authentification réussie", visible=False)
+)
 
 # Créer une instance de HfFileSystem
 fs = HfFileSystem()
@@ -28,35 +45,6 @@ def list_repo_content(repo_path):
     content = repo_info["subdirectories"] + repo_info["files"]
     return content
 
-def create_subfolder(repo_path, subfolder_name):
-    repo_id = repo_path.split("/")[0]
-    subfolder_path = f"{repo_path}/{subfolder_name}"
-    fs.mkdir(subfolder_path)
-    return f"Sous-répertoire {subfolder_path} créé avec succès."
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        repo_path = gr.Textbox(label="Chemin du dépôt")
-        list_button = gr.Button("Lister le contenu")
-        create_button = gr.Button("Créer un sous-répertoire")
-        subfolder_name = gr.Textbox(label="Nom du sous-répertoire")
-
-# Récupérer token depuis les variables d'environnement
-hf_token = os.getenv("MisterAI_bigscience_bloom_560m")
-
-# Configurer le token pour l'utilisation avec Hugging Face
-if hf_token:
-    HfFolder.save_token(hf_token)
-else:
-    raise ValueError("Le token Hugging Face n'est pas configuré. Assurez-vous qu'il est défini dans les variables d'environnement.")
-
-def list_repo_content(repo_path):
-    api = HfApi()
-    repo_id = repo_path.split("/")[0]
-    repo_info = api.repo_info(repo_id)
-    content = repo_info["subdirectories"] + repo_info["files"]
-    return content
-
 def create_subfolder(repo_path, subfolder_name):
     api = HfApi()
     repo_id = repo_path.split("/")[0]
@@ -64,6 +52,8 @@ def create_subfolder(repo_path, subfolder_name):
     api.create_folder(repo_id, subfolder_path)
     return f"Sous-répertoire {subfolder_path} créé avec succès."
 
+
+
 with gr.Blocks() as demo:
     with gr.Row():
         repo_path = gr.Textbox(label="Chemin du dépôt")
@@ -87,27 +77,7 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
+    auth_demo.launch()
     demo.launch()
 
-    with gr.Row():
-        repo_content = gr.Textbox(label="Contenu du dépôt")
-
-    list_button.click(
-        list_repo_content,
-        inputs=repo_path,
-        outputs=repo_content
-    )
-
-    create_button.click(
-        create_subfolder,
-        inputs=[repo_path, subfolder_name],
-        outputs=repo_content
-    )
-
-if __name__ == "__main__":
-    demo.launch()
-
-
-
-