Update app.py

app.py

@@ -120,137 +120,7 @@ async def download_post(request: PostRequest):
     return {"download_url": download_url}
 
 
-# Directly embed session data
-SESSION_DATA = {
-    "sessionid": "49379440378%3AAMKUUgo79iWhmt%3A1%3AAYcuPYtZFJAwzPlT0u_xuXkaiBHbDJ-8igmXkPBmzQ",
-    "mid": "Z07d8AALAAHe7OnMarRANiQOAt7a",
-    "ig_pr": "1",
-    "ig_vw": "1920",
-    "ig_cb": "1",
-    "csrftoken": "xNZpGQiAKQkf4IU2E8UAV0snSGMZEPCt",
-    "s_network": "",
-    "ds_user_id": "49379440378",
-    "ig_did": "0DCC4ACA-6794-4778-88D8-573410BD5CC4"
-}
 
-app = FastAPI()
-
-class DownloadRequest(BaseModel):
-    url: str
-
-class DownloadResponse(BaseModel):
-    success: bool
-    message: str
-    urls: Optional[List[str]] = None
-    type: Optional[str] = None
-    shortcode: Optional[str] = None
-
-class StoryRequest(BaseModel):
-    username: Optional[str] = None
-    url: Optional[str] = None
-
-class StoryResponse(BaseModel):
-    success: bool
-    message: str
-    urls: Optional[List[str]] = None
-
-def get_authenticated_loader():
-    """Create and return an authenticated Instaloader instance"""
-    loader = instaloader.Instaloader(
-        download_videos=True,
-        download_video_thumbnails=False,
-        download_geotags=False,
-        download_comments=False,
-        save_metadata=False,
-        compress_json=False,
-        sleep=True,
-        quiet=True
-    )
-    
-    try:
-        # Set the session data directly
-        loader.context._session.cookies.update(SESSION_DATA)
-        
-        # Verify session is valid
-        test_profile = instaloader.Profile.from_username(loader.context, INSTAGRAM_USERNAME)
-        _ = test_profile.userid
-        logger.info("Successfully loaded and verified session")
-        return loader
-        
-    except Exception as e:
-        logger.error(f"Failed to load session: {str(e)}")
-        raise HTTPException(
-            status_code=500,
-            detail="Failed to authenticate with Instagram"
-        )
-
-@app.post("/api/story", response_model=StoryResponse)
-async def download_story(request: StoryRequest):
-    try:
-        # Extract username from request
-        if request.url:
-            username = request.url.split('/')[-1].split('?')[0]
-        elif request.username:
-            username = request.username
-        else:
-            raise HTTPException(status_code=400, detail="Username or URL is required")
-
-        # Add small delay to avoid rate limiting
-        time.sleep(random.uniform(1, 3))
-
-        # Get authenticated loader
-        loader = get_authenticated_loader()
-        
-        try:
-            # Fetch profile
-            profile = instaloader.Profile.from_username(loader.context, username)
-            
-            # Add small delay before fetching stories
-            time.sleep(random.uniform(2, 4))
-            
-            # Get stories
-            story_items = loader.get_stories(userids=[profile.userid])
-            
-            urls = []
-            story_item_count = 0
-            
-            for story in story_items:
-                for item in story.get_items():
-                    story_item_count += 1
-                    if item.is_video:
-                        urls.append(item.video_url)
-                    else:
-                        urls.append(item.url)
-
-            if story_item_count == 0:
-                return StoryResponse(
-                    success=True,
-                    message="No active stories found for this user",
-                    urls=[]
-                )
-
-            return StoryResponse(
-                success=True,
-                message=f"Retrieved {story_item_count} stories successfully",
-                urls=urls
-            )
-
-        except instaloader.exceptions.ProfileNotExistsException:
-            raise HTTPException(status_code=404, detail="Instagram profile not found")
-            
-        except Exception as e:
-            if "429" in str(e):
-                raise HTTPException(
-                    status_code=429,
-                    detail="Rate limited by Instagram. Please try again in 15 minutes."
-                )
-            raise
-
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Story download error: {str(e)}")
-        raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/api/health")
 async def health_check():