Upload folder using huggingface_hub

utils.py

@@ -5,12 +5,13 @@ import requests
 
 def extract_instagram_code(url: str) -> str:
     parts = [part for part in url.split('/') if part and '?' not in part]
-    # The post code is expected to be the part after 'p'
-    if 'p' in parts:
-        post_code_index = parts.index('p') + 1
-        if post_code_index < len(parts):
-            return parts[post_code_index]
-    return "Invalid URL or post code not found"
+    # The code is expected to be the part after 'p' or 'reel'
+    for key in ['p', 'reel']:
+        if key in parts:
+            code_index = parts.index(key) + 1
+            if code_index < len(parts):
+                return parts[code_index]
+    return "Invalid URL or code not found"
 
 
 def get_users_from_keyword(keyword: str):