Update app.py

app.py

@@ -31,9 +31,13 @@ def fetch_manifest(image, tag):
         return None
 
 def upload_to_huggingface(repo_id, folder_path, oauth_token: Union[gr.OAuthToken, None]):
+
     token = oauth_token.token if oauth_token else None
+    
     api = HfApi(token=token)
+    
     repo_path = api.create_repo(repo_id=repo_id, repo_type="model", exist_ok=True)
+    
     print(f"Repo created {repo_path}")
     try:
         api.upload_folder(
@@ -41,7 +45,7 @@ def upload_to_huggingface(repo_id, folder_path, oauth_token: Union[gr.OAuthToken
             repo_id=repo_id,
             repo_type="model",
         )
-        return "Upload successful"
+        return "Upload successful", repo_path
     except Exception as e:
         return f"Upload failed: {str(e)}"
 
@@ -55,7 +59,7 @@ def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None
         return f"Failed to fetch the manifest for {image}:{tag}"
 
     # Save the manifest JSON to the blobs folder
-    manifest_file_path = "blobs/manifest.json"
+    manifest_file_path = "blobs/manifest"
     os.makedirs(os.path.dirname(manifest_file_path), exist_ok=True)
     with open(manifest_file_path, 'w') as f:
         json.dump(manifest_json, f)
@@ -73,16 +77,16 @@ def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None
         download_file(config_digest, image)
     try:
         # Upload to Hugging Face Hub
-        upload_result = upload_to_huggingface(repo_id, 'blobs', oauth_token)
+        upload_result, repo_path = upload_to_huggingface(repo_id, 'blobs', oauth_token)
     
         # Delete the blobs folder
         shutil.rmtree('blobs')
 
-        return f"Successfully fetched and downloaded files for {image}:{tag}\n{upload_result}\nBlobs folder deleted"
+        return f"Successfully fetched and downloaded * uploaded files for {image}:{tag} <a href=\'{repo_path}\' target="_blank" style="text-decoration:underline">here</a> 🦙"
     except Exception as e:
         shutil.rmtree('blobs', ignore_errors=True)
 
-        return f"Error found: {str(e)}"
+        return f"We got an error, my dude, here's what the error looks like: {str(e)}"
 
 css = """
 .main_ui_logged_out{opacity: 0.3; pointer-events: none}
@@ -98,7 +102,7 @@ with gr.Blocks() as demo:
     image_tag_input = gr.Textbox(placeholder="Enter Ollama ID", label="Image and Tag")
     repo_id_input = gr.Textbox(placeholder="Enter Hugging Face repo ID", label="Hugging Face Repo ID")
     
-    result_output = gr.Textbox(label="Result")
+    result_output = gr.Markdown(label="Result")
     
     process_button = gr.Button("Process")
     process_button.click(fn=process_image_tag, inputs=[image_tag_input, repo_id_input], outputs=result_output)