Update app.py (#13)

- Update app.py (117597ab20fab226e8cff281516e6bf87c9e1c7e)

app.py

@@ -47,40 +47,41 @@ def upload_to_huggingface(repo_id, folder_path, oauth_token: Union[gr.OAuthToken
 
 def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None]):
     # Extract image and tag from the input
+    image, tag = image_tag.split(':')
+
+    # Fetch the manifest JSON
+    manifest_json = fetch_manifest(image, tag)
+    if not manifest_json or 'errors' in manifest_json:
+        return f"Failed to fetch the manifest for {image}:{tag}"
+
+    # Save the manifest JSON to the blobs folder
+    manifest_file_path = "blobs/manifest.json"
+    os.makedirs(os.path.dirname(manifest_file_path), exist_ok=True)
+    with open(manifest_file_path, 'w') as f:
+        json.dump(manifest_json, f)
+
+    # Extract the digest values from the JSON
+    digests = [layer['digest'] for layer in manifest_json.get('layers', [])]
+
+    # Download each file
+    for digest in digests:
+        download_file(digest, image)
+
+    # Download the config file
+    config_digest = manifest_json.get('config', {}).get('digest')
+    if config_digest:
+        download_file(config_digest, image)
     try:
-        image, tag = image_tag.split(':')
-    
-        # Fetch the manifest JSON
-        manifest_json = fetch_manifest(image, tag)
-        if not manifest_json or 'errors' in manifest_json:
-            return f"Failed to fetch the manifest for {image}:{tag}"
-    
-        # Save the manifest JSON to the blobs folder
-        manifest_file_path = "blobs/manifest.json"
-        os.makedirs(os.path.dirname(manifest_file_path), exist_ok=True)
-        with open(manifest_file_path, 'w') as f:
-            json.dump(manifest_json, f)
-    
-        # Extract the digest values from the JSON
-        digests = [layer['digest'] for layer in manifest_json.get('layers', [])]
-    
-        # Download each file
-        for digest in digests:
-            download_file(digest, image)
-    
-        # Download the config file
-        config_digest = manifest_json.get('config', {}).get('digest')
-        if config_digest:
-            download_file(config_digest, image)
-    
         # Upload to Hugging Face Hub
         upload_result = 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"
     except Exception as e:
         shutil.rmtree('blobs', ignore_errors=True)
+
         return f"Error found: {str(e)}"
 
 css = """