Update app.py

app.py

@@ -1,13 +1,9 @@
 import logging
+from dotenv import load_dotenv
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from fastapi.responses import StreamingResponse
 import instaloader
-import os
-from dotenv import load_dotenv
-from typing import Optional, List
-import time
-import random
-from fastapi.middleware.cors import CORSMiddleware
+import requests
 
 # Load environment variables
 load_dotenv()
@@ -20,120 +16,56 @@ INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Check if credentials are available
-if not INSTAGRAM_USERNAME or not INSTAGRAM_PASSWORD:
-    logger.error("Instagram credentials are not set in the environment variables.")
-    raise RuntimeError("Instagram credentials missing.")
-
-app = FastAPI(title="Instagram Post Downloader")
-
-# Add CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
 
-class PostRequest(BaseModel):
-    url: str
+app = FastAPI(title="Instagram Stories API", docs_url=None, redoc_url=None)
 
-class DownloadResponse(BaseModel):
-    download_urls: List[str]
-
-def extract_shortcode(url: str) -> str:
-    """Extract Instagram post shortcode from URL"""
-    url = url.strip()
-    if 'instagram.com' not in url:
-        raise ValueError("Invalid Instagram URL")
-    parts = url.split('/')
-    for i, part in enumerate(parts):
-        if part in ['p', 'reel', 'tv'] and (i + 1) < len(parts):
-            shortcode = parts[i + 1].split('?')[0]
-            return shortcode
-    if 'instagram.com/p/' in url:
-        return url.split('/p/')[1].split('/')[0].split('?')[0]
-    raise ValueError("Invalid Instagram URL")
-
-
-@app.post("/dl", response_model=DownloadResponse)
-async def download_post(request: PostRequest):
-    """Download Instagram post and return media URL"""
+@app.get("/stories/{username}", summary="Get available stories")
+async def get_stories(username: str):
+    """Retrieve metadata for public Instagram stories"""
+    L = instaloader.Instaloader()
     try:
-        shortcode = extract_shortcode(request.url)
-        logger.info(f"Extracted shortcode: {shortcode}")
-    except ValueError as e:
-        logger.error(f"Invalid URL: {request.url}")
-        raise HTTPException(status_code=400, detail=str(e))
-
+        profile = instaloader.Profile.from_username(L.context, username)
+    except instaloader.exceptions.ProfileNotExistsException:
+        raise HTTPException(404, "Profile not found")
+    
+    stories = [{
+        "id": str(item.mediaid),
+        "url": item.url,
+        "type": "video" if item.is_video else "image",
+        "timestamp": item.date_utc.isoformat(),
+        "views": item.view_count
+    } for story in profile.get_stories() for item in story.get_items()]
+    
+    return {"data": stories} if stories else HTTPException(404, "No stories found")
+
+@app.get("/download/{media_id}", summary="Download story content")
+async def download_story(media_id: str, username: str):
+    """Download specific story media by ID"""
     L = instaloader.Instaloader()
-    session_file = f"{INSTAGRAM_USERNAME}.session"
-
-    if os.path.exists(session_file):
-        try:
-            L.load_session_from_file(INSTAGRAM_USERNAME, session_file)
-            logger.info("Loaded session from file.")
-        except FileNotFoundError:
-            pass
-    else:
-        try:
-            # Login to Instagram
-            L.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
-            logger.info("Successfully logged in to Instagram")
-            L.save_session_to_file(session_file)
-            logger.info("Saved session to file.")
-        except instaloader.exceptions.InstaloaderException as e:
-            logger.error(f"Login error: {e}")
-            raise HTTPException(status_code=500, detail="Error logging in to Instagram")
-
-    # Configure retry parameters
-    max_retries = 3
-    retry_delay = random.uniform(1, 3)
-
-    for attempt in range(max_retries):
-        try:
-            # Fetch post by shortcode
-            post = instaloader.Post.from_shortcode(L.context, shortcode)
-            logger.info(f"Fetched post: {post}")
-            break
-        except instaloader.exceptions.ConnectionException as e:
-            logger.error(f"Connection error (attempt {attempt + 1}): {e}")
-            if attempt == max_retries - 1:
-                raise HTTPException(status_code=500, detail="Failed to connect to Instagram after multiple attempts")
-            time.sleep(retry_delay)
-            retry_delay *= 2  # Exponential backoff
-        except instaloader.exceptions.InvalidArgumentException:
-            logger.error(f"Post not found for shortcode: {shortcode}")
-            raise HTTPException(status_code=404, detail="Post not found")
-        except instaloader.exceptions.InstaloaderException as e:
-            logger.error(f"Instaloader error: {e}")
-            raise HTTPException(status_code=500, detail="Error fetching post")
-        except Exception as e:
-            logger.error(f"Unexpected error: {e}")
-            raise HTTPException(status_code=500, detail="Internal server error")
-
-    download_urls = []
-    if post.typename == 'GraphSidecar':
-        # Handle carousel posts
-        for node in post.get_sidecar_nodes():
-            if node.is_video:
-                download_urls.append(node.video_url)
-            else:
-                download_urls.append(node.display_url)
-    else:
-        if post.is_video:
-            download_urls.append(post.video_url)
-        else:
-            download_urls.append(post.url)
-
-    if not download_urls:
-        logger.error("No media found in post")
-        raise HTTPException(status_code=404, detail="No media found in post")
+    try:
+        profile = instaloader.Profile.from_username(L.context, username)
+    except instaloader.exceptions.ProfileNotExistsException:
+        raise HTTPException(404, "Profile not found")
 
-    return {"download_urls": download_urls}
+    # Find the story item
+    story_item = next((item for story in profile.get_stories() 
+                      for item in story.get_items() 
+                      if str(item.mediaid) == media_id), None)
 
+    if not story_item:
+        raise HTTPException(404, "Story not found")
 
-@app.get("/api/health")
-async def health_check():
-    return {"status": "healthy"}
+    try:
+        response = requests.get(story_item.url, stream=True)
+        response.raise_for_status()
+    except requests.exceptions.RequestException:
+        raise HTTPException(500, "Failed to fetch media")
+
+    return StreamingResponse(
+        response.iter_content(chunk_size=8192),
+        media_type="video/mp4" if story_item.is_video else "image/jpeg",
+        headers={
+            "Content-Disposition": f"attachment; filename={username}_{media_id}"
+            f".{'mp4' if story_item.is_video else 'jpg'}"
+        }
+    )