Update app.py

app.py

@@ -13,6 +13,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 def process_model(
     model_id: str,
     file_path: str,
+    file_path_dropdown: str,
     key: str,
     value: str,
     oauth_token: gr.OAuthToken | None,
@@ -24,6 +25,11 @@ def process_model(
 
     MODEL_NAME = model_id.split("/")[-1]
 
+    if file_path_dropdown:
+        FILE_PATH = file_path_dropdown
+    else:
+        FILE_PATH = file_path
+
     FILE_NAME = file_path.split("/")[-1]
 
     api.snapshot_download(
@@ -33,7 +39,7 @@ def process_model(
     )
     print("Model downloaded successully!")
 
-    metadata_update = f"python3 llama.cpp/gguf-py/scripts/gguf_set_metadata.py {MODEL_NAME}/{file_path} {key} {value}"
+    metadata_update = f"yes | python3 llama.cpp/gguf-py/scripts/gguf_set_metadata.py {MODEL_NAME}/{FILE_PATH} {key} {value}"
     subprocess.run(metadata_update, shell=True)
     print(f"Model metadata {key} updated to {value} successully!")
 
@@ -60,13 +66,15 @@ with gr.Blocks() as demo:
 
     file_path = gr.Textbox(lines=1, label="File path")
 
+    file_path_dropdown = gr.Dropdown(["None"], label="File", visible=False)
+
     key = gr.Textbox(lines=1, label="Key")
 
     value = gr.Textbox(lines=1, label="Value")
 
     iface = gr.Interface(
         fn=process_model,
-        inputs=[model_id, file_path, key, value],
+        inputs=[model_id, file_path, file_path_dropdown, key, value],
         outputs=[
             gr.Markdown(label="output"),
             gr.Image(show_label=False),
@@ -76,6 +84,24 @@ with gr.Blocks() as demo:
         api_name=False,
     )
 
+    def updateFilePath(model_id: HuggingfaceHubSearch):
+        try:
+            api = HfApi()
+            files = []
+            for file in api.list_repo_tree(
+                repo_id=model_id,
+                recursive=True,
+            ):
+                files.append(file.path)
+
+            return gr.update(visible=False), gr.update(visible=True, choices=files)
+        except Exception:
+            return gr.update(visible=True), gr.update(visible=False)
+
+    model_id.change(
+        fn=updateFilePath, inputs=model_id, outputs=[file_path, file_path_dropdown]
+    )
+
 
 def restart_space():
     HfApi().restart_space(