Update app.py

app.py

@@ -389,17 +389,19 @@ def extract_video_info(video_url: str) -> str:
                 else:
                     new_result = result
                     # Check if new_result has more than one item
-                    if len(new_result) > 1:
-                        # Get the last item
+                    if len(new_result) > 3:
+                        for i in range(3, len(new_result)):
+                            item = new_result[i]
+                            item["format_id"] = f'{item["format_id"]} - Chrunos Shortcuts Premium Only'
+                            item["url"] = 'https://chrunos.com/premium-shortcuts/'
+                    elif 2 <= len(new_result) <= 3:
                         last_item = new_result[-1]
-                        # Modify the format_id
                         last_item["format_id"] = f'{last_item["format_id"]} - Chrunos Shortcuts Premium Only'
-                        # Modify the url
                         last_item["url"] = 'https://chrunos.com/premium-shortcuts/'
                     elif len(new_result) == 1:
                         new_item = {"url": "https://chrunos.com/premium-shortcuts/",
-                         "format_id": "Best Qaulity Video - Chrunos Shortcuts Premium Only"                            
-                        }
+                                    "format_id": "Best Qaulity Video - Chrunos Shortcuts Premium Only"
+                                    }
                         new_result.append(new_item)
 
                     return new_result
@@ -426,6 +428,16 @@ def extract_video_info(video_url: str) -> str:
 
 @app.post("/test")
 async def test_download(request: Request):
+    user_ip = get_user_ip(request)
+    if rate_limiter.is_rate_limited(user_ip):
+        current_count = rate_limiter.get_current_count(user_ip)
+        raise HTTPException(
+            status_code=429,
+            detail={
+                "error": "You have exceeded the maximum number of requests per day. Please try again tomorrow.",
+                "url": "https://t.me/chrunoss"
+            }
+        )
     data = await request.json()
     video_url = data.get('url')
     response = extract_video_info(video_url)