Update app.py

app.py

@@ -1,110 +1,23 @@
-# main.py
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-import instaloader
-import os
-from dotenv import load_dotenv
-import re
-from typing import Optional, List
-
-# Load environment variables
-load_dotenv()
-
-# Get credentials from environment variables
-INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
-INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
-
-app = FastAPI(title="Instagram Downloader API")
-
-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
-
-def clean_url(url: str) -> Optional[str]:
-    """Clean Instagram URL and extract shortcode"""
-    try:
-        cleaned_url = url.split('?')[0].split('#')[0].strip('/')
-        patterns = [
-            r'instagram.com/reel/([A-Za-z0-9_-]+)',
-            r'instagram.com/p/([A-Za-z0-9_-]+)',
-            r'instagram.com/stories/([A-Za-z0-9_-]+)',
-            r'/([A-Za-z0-9_-]{11})[/]?$'
-        ]
-        
-        for pattern in patterns:
-            match = re.search(pattern, cleaned_url)
-            if match:
-                return match.group(1)
-        
-        segments = cleaned_url.split('/')
-        if segments:
-            return segments[-1]
-                
-        return None
-    except Exception:
-        return None
-
-@app.post("/api/download", response_model=DownloadResponse)
-async def download_media(request: DownloadRequest):
-    try:
-        # Extract shortcode
-        shortcode = clean_url(request.url)
-        if not shortcode:
-            raise HTTPException(status_code=400, detail="Invalid URL format")
-
-        # Initialize loader
-        loader = instaloader.Instaloader(
-            download_videos=True,
-            download_video_thumbnails=False,
-            download_geotags=False,
-            download_comments=False,
-            save_metadata=False,
-            compress_json=False
-        )
-
-        # Login
-        if not all([INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD]):
-            raise HTTPException(status_code=500, detail="Missing Instagram credentials")
-        
-        loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
-
-        # Get post
-        post = instaloader.Post.from_shortcode(loader.context, shortcode)
-        
-        urls = []
-        
-        if post.is_video:
-            media_type = "video"
-            urls.append(post.video_url)
-        else:
-            media_type = "image"
-            if post.typename == "GraphSidecar":
-                # Multiple images
-                for node in post.get_sidecar_nodes():
-                    urls.append(node.display_url)
-            else:
-                # Single image
-                urls.append(post.url)
-
-        return DownloadResponse(
-            success=True,
-            message="Media URLs retrieved successfully",
-            urls=urls,
-            type=media_type,
-            shortcode=shortcode
-        )
-
-    except instaloader.exceptions.InstaloaderException as e:
-        raise HTTPException(status_code=400, detail=str(e))
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
-@app.get("/api/health")
-async def health_check():
-    return {"status": "healthy"}
+# app.py
+import streamlit as st
+from fastapi import FastAPI
+import uvicorn
+from api import app as api_app
+import threading
+
+# Your existing Streamlit app code here
+def run_streamlit():
+    # Your existing Streamlit code...
+    st.title('Instagram Downloader')
+    # ... rest of your Streamlit code ...
+
+def run_api():
+    uvicorn.run(api_app, host="0.0.0.0", port=8000)
+
+if __name__ == "__main__":
+    # Run API in a separate thread
+    api_thread = threading.Thread(target=run_api, daemon=True)
+    api_thread.start()
+    
+    # Run Streamlit
+    run_streamlit()