Update app.py

app.py

@@ -9,10 +9,18 @@ from apscheduler.schedulers.background import BackgroundScheduler
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-api = HfApi()
 
+def process_model(
+    model_id: str,
+    file_path: str,
+    key: str,
+    value: str,
+    oauth_token: gr.OAuthToken | None,
+):
+    if oauth_token.token is None:
+        raise ValueError("You must be logged in to use gguf-metadata-updater")
 
-def process_model(model_id: str, file_path: str, key: str, value: str, hf_token):
+    api = HfApi(token=oauth_token.token)
 
     MODEL_NAME = model_id.split("/")[-1]
 
@@ -33,29 +41,40 @@ def process_model(model_id: str, file_path: str, key: str, value: str, hf_token)
     api.upload_folder(
         folder_path=MODEL_NAME,
         repo_id=model_id,
-        allow_patterns=["*.gguf", "$.md"],
-        token=hf_token,
+        allow_patterns=["*.gguf"],
     )
     print("Uploaded successfully!")
 
     return "Processing complete."
 
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=process_model,
-    inputs=[
-        gr.Textbox(lines=1, label="Model ID"),
-        gr.Textbox(lines=1, label="File path"),
-        gr.Textbox(lines=1, label="Key"),
-        gr.Textbox(lines=1, label="Value"),
-        gr.Textbox(lines=1, label="Token"),
-    ],
-    outputs="text",
-)
+with gr.Blocks() as demo:
+    gr.Markdown("You must be logged in to use GGUF metadata updated.")
+    gr.LoginButton(min_width=250)
 
-# Launch the interface
-iface.launch(debug=True)
+    model_id = HuggingfaceHubSearch(
+        label="Hub Model ID",
+        placeholder="Search for model id on Huggingface",
+        search_type="model",
+    )
+
+    file_path = gr.Textbox(lines=1, label="File path")
+
+    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],
+        outputs=[
+            gr.Markdown(label="output"),
+            gr.Image(show_label=False),
+        ],
+        title="Update metadata for a GGUF file",
+        description="The space takes an HF repo, a file within that repo, a metadata key, and new metadata value to update it to.",
+        api_name=False,
+    )
 
 
 def restart_space():