Update app.py

app.py

@@ -1,5 +1,7 @@
 from fastapi import FastAPI, HTTPException, status  
+from fastapi.responses import StreamingResponse  
 import instaloader  
+import requests  
 import os  
 import time  
 import logging  
@@ -169,13 +171,26 @@ async def get_stories(username: str):
         try:  
             for story in L.get_stories(userids=[profile.userid]):  
                 for item in story.get_items():  
-                    stories.append({  
+                    # Create a story dict with safe attribute access  
+                    story_data = {  
                         "id": str(item.mediaid),  
                         "url": item.url,  
                         "type": "video" if item.is_video else "image",  
                         "timestamp": item.date_utc.isoformat(),  
-                        "views": item.view_count if item.is_video else None,  
-                    })  
+                    }  
+                    
+                    # Only try to add view_count if it's a video and the attribute exists  
+                    if item.is_video:  
+                        try:  
+                            # Safely check if view_count attribute exists  
+                            if hasattr(item, 'view_count'):  
+                                story_data["views"] = item.view_count  
+                        except AttributeError:  
+                            # Skip adding views if the attribute doesn't exist  
+                            pass  
+                    
+                    stories.append(story_data)  
+                    
         except instaloader.exceptions.QueryReturnedNotFoundException:  
             logger.error("Stories not found")  
             raise HTTPException(  
@@ -198,4 +213,67 @@ async def get_stories(username: str):
 
     except Exception as e:  
         logger.error(f"Critical failure: {str(e)}")  
-        raise
+        raise  
+
+@app.get("/download/{url:path}")  
+@handle_instagram_errors  
+async def download_media(url: str):  
+    """Download and proxy media content"""  
+    logger.info(f"Download request for: {url}")  
+    
+    try:  
+        # Basic validation to prevent arbitrary URL access  
+        if not url.startswith(("https://instagram", "https://scontent")):  
+            raise HTTPException(  
+                status_code=status.HTTP_400_BAD_REQUEST,  
+                detail="Invalid URL format"  
+            )  
+            
+        # Random delay to avoid detection patterns  
+        time.sleep(random.uniform(1.0, 3.0))  
+        
+        # Configure headers to mimic a browser  
+        headers = {  
+            "User-Agent": random.choice(USER_AGENTS),  
+            "Accept": "*/*",  
+            "Accept-Language": "en-US,en;q=0.9",  
+            "Referer": "https://www.instagram.com/",  
+            "Sec-Fetch-Dest": "empty",  
+            "Sec-Fetch-Mode": "cors",  
+            "Sec-Fetch-Site": "cross-site",  
+        }  
+        
+        # Request the media with a session  
+        response = requests.get(url, headers=headers, stream=True, timeout=30)  
+        response.raise_for_status()  
+        
+        # Determine content type from response or URL  
+        if "Content-Type" in response.headers:  
+            content_type = response.headers["Content-Type"]  
+        elif url.endswith((".jpg", ".jpeg")):  
+            content_type = "image/jpeg"  
+        elif url.endswith(".mp4"):  
+            content_type = "video/mp4"  
+        else:  
+            content_type = "application/octet-stream"  
+            
+        logger.info(f"Media downloaded successfully: {content_type}")  
+        
+        # Stream the response back  
+        return StreamingResponse(  
+            response.iter_content(chunk_size=8192),  
+            media_type=content_type,  
+            headers={  
+                "Content-Disposition": f"attachment; filename={url.split('/')[-1]}",  
+                "Cache-Control": "no-cache, no-store, must-revalidate",  
+                "Pragma": "no-cache",  
+                "Expires": "0",  
+            }  
+        )  
+        
+    except requests.exceptions.RequestException as e:  
+        logger.error(f"Download failed: {str(e)}")  
+        raise HTTPException(  
+            status_code=status.HTTP_502_BAD_GATEWAY,  
+            detail="Failed to download media"  
+        )