inst

Running

App Files Files Community

Chrunos commited on Apr 8

Commit

d27a73d

verified ·

1 Parent(s): 226ea29

Update app.py

Browse files

Files changed (1) hide show

app.py +152 -309

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, status, Request, Form
 from fastapi.responses import StreamingResponse, HTMLResponse, RedirectResponse
 import requests
 import os
@@ -7,8 +7,6 @@ import logging
 import random
 import json
 from datetime import datetime, timedelta
-import re
-from typing import Optional
 # Configure logging
 logging.basicConfig(
@@ -39,174 +37,7 @@ USER_AGENTS = [
     "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
 ]
-# Cookie import HTML template
-COOKIE_IMPORT_HTML = """
-<!DOCTYPE html>
-<html>
-<head>
-    <title>Import Instagram Session</title>
-    <style>
-        body { font-family: Arial; max-width: 700px; margin: 0 auto; padding: 20px; }
-        .container { background-color: #fff; border-radius: 8px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); padding: 25px; }
-        h1 { color: #262626; text-align: center; }
-        .form-group { margin-bottom: 15px; }
-        label { display: block; margin-bottom: 10px; }
-        textarea { width: 100%; padding: 10px; border: 1px solid #dbdbdb; border-radius: 3px; height: 150px; }
-        .btn { background-color: #0095f6; border: none; color: white; padding: 10px 15px; border-radius: 4px; cursor: pointer; font-weight: bold; }
-        .btn-block { width: 100%%; }
-        .status { margin-top: 20px; padding: 10px; border-radius: 4px; }
-        .success { background-color: #e8f5e9; color: #388e3c; }
-        .error { background-color: #ffebee; color: #d32f2f; }
-        .instructions { background-color: #f8f9fa; padding: 15px; border-radius: 4px; margin-bottom: 20px; }
-        .code { font-family: monospace; background-color: #f1f1f1; padding: 2px 5px; border-radius: 3px; }
-        ol { padding-left: 20px; }
-        li { margin-bottom: 10px; }
-        .tabs { display: flex; margin-bottom: 20px; }
-        .tab { padding: 10px 15px; cursor: pointer; background-color: #f8f9fa; border-radius: 4px 4px 0 0; margin-right: 5px; }
-        .tab.active { background-color: #0095f6; color: white; }
-        .tab-content { display: none; }
-        .tab-content.active { display: block; }
-    </style>
-</head>
-<body>
-    <div class="container">
-        <h1>Import Instagram Session</h1>
-        <div class="tabs">
-            <div class="tab active" onclick="switchTab('tab1', this)">Cookie Format</div>
-            <div class="tab" onclick="switchTab('tab2', this)">JSON Format</div>
-        </div>
-        <div id="tab1" class="tab-content active">
-            <div class="instructions">
-                <h3>How to get your Instagram cookies:</h3>
-                <ol>
-                    <li>Log in to Instagram in your browser</li>
-                    <li>Open browser DevTools (F12 or right-click and select "Inspect")</li>
-                    <li>Go to the "Application" or "Storage" tab</li>
-                    <li>Under "Cookies", find "instagram.com"</li>
-                    <li>Find the <span class="code">sessionid</span> cookie and copy its value</li>
-                    <li>Also copy the <span class="code">ds_user_id</span> and <span class="code">csrftoken</span> cookies if available</li>
-                    <li>Paste the cookies in the format: <span class="code">name=value; name2=value2;</span></li>
-                </ol>
-            </div>
-            <form method="post" action="/import-cookies">
-                <div class="form-group">
-                    <label for="cookies">Paste your Instagram cookies here:</label>
-                    <textarea id="cookies" name="cookies" placeholder="sessionid=YOUR_SESSION_ID; ds_user_id=YOUR_USER_ID; csrftoken=YOUR_CSRF_TOKEN;" required></textarea>
-                </div>
-                <div class="form-group">
-                    <label for="username">Your Instagram username:</label>
-                    <input type="text" id="username" name="username" style="width: 100%; padding: 10px; border: 1px solid #dbdbdb; border-radius: 3px;" required>
-                </div>
-                <button type="submit" class="btn btn-block">Import Session</button>
-            </form>
-        </div>
-        <div id="tab2" class="tab-content">
-            <div class="instructions">
-                <h3>How to get your cookies as JSON:</h3>
-                <ol>
-                    <li>Install a browser extension like "EditThisCookie" or "Cookie-Editor"</li>
-                    <li>Log in to Instagram in your browser</li>
-                    <li>Click on the extension icon</li>
-                    <li>Choose "Export" or similar option to copy all cookies</li>
-                    <li>Paste the JSON data below</li>
-                </ol>
-            </div>
-            <form method="post" action="/import-cookies-json">
-                <div class="form-group">
-                    <label for="cookies_json">Paste your Instagram cookies JSON here:</label>
-                    <textarea id="cookies_json" name="cookies_json" placeholder='[{"name":"sessionid","value":"YOUR_SESSION_ID","domain":".instagram.com"}]' required></textarea>
-                </div>
-                <div class="form-group">
-                    <label for="username_json">Your Instagram username:</label>
-                    <input type="text" id="username_json" name="username" style="width: 100%; padding: 10px; border: 1px solid #dbdbdb; border-radius: 3px;" required>
-                </div>
-                <button type="submit" class="btn btn-block">Import Session</button>
-            </form>
-        </div>
-        %s
-    </div>
-    <script>
-        function switchTab(tabId, tabElement) {
-            // Hide all tab contents
-            document.querySelectorAll('.tab-content').forEach(content => {
-                content.classList.remove('active');
-            });
-            // Deactivate all tabs
-            document.querySelectorAll('.tab').forEach(tab => {
-                tab.classList.remove('active');
-            });
-            // Activate clicked tab and its content
-            document.getElementById(tabId).classList.add('active');
-            tabElement.classList.add('active');
-        }
-    </script>
-</body>
-</html>
-"""
-# Status page HTML
-STATUS_HTML = """
-<!DOCTYPE html>
-<html>
-<head>
-    <title>API Status</title>
-    <style>
-        body { font-family: Arial; max-width: 600px; margin: 0 auto; padding: 20px; }
-        .container { background-color: #fff; border-radius: 8px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); padding: 25px; }
-        h1 { color: #262626; text-align: center; }
-        .status { display: inline-block; padding: 5px 10px; border-radius: 4px; margin-left: 10px; font-weight: bold; }
-        .connected { background-color: #e8f5e9; color: #388e3c; }
-        .disconnected { background-color: #ffebee; color: #d32f2f; }
-        .info { background-color: #f5f5f5; border-radius: 4px; padding: 15px; margin: 15px 0; }
-        .btn { background-color: #0095f6; border: none; color: white; padding: 10px 15px; border-radius: 4px; text-decoration: none; display: inline-block; margin-right: 10px; }
-        .btn-danger { background-color: #f44336; }
-        table { width: 100%%; border-collapse: collapse; margin-top: 15px; }
-        th, td { text-align: left; padding: 8px; border-bottom: 1px solid #ddd; }
-    </style>
-</head>
-<body>
-    <div class="container">
-        <h1>Instagram API Status</h1>
-        <h2>Connection Status
-            %s
-        </h2>
-        <div class="info">
-            %s
-        </div>
-        <a href="/import" class="btn">%s</a>
-        %s
-        <h2>API Endpoints</h2>
-        <table>
-            <tr><th>Endpoint</th><th>Description</th></tr>
-            <tr><td><code>/stories/{username}</code></td><td>Get stories for a user</td></tr>
-            <tr><td><code>/download/{url}</code></td><td>Download media from a URL</td></tr>
-            <tr><td><code>/status</code></td><td>View API status</td></tr>
-        </table>
-        <h2>Cache Info</h2>
-        <div class="info">
-            <p>Cached users: %d</p>
-            <p>Total stories cached: %d</p>
-        </div>
-    </div>
-</body>
-</html>
-"""
-# Functions for cache and Instagram session
 def get_cache_key(username):
     return username.lower()
@@ -250,6 +81,7 @@ def should_rate_limit_request(username):
     LAST_REQUEST[key] = now
     return False
 def get_instagram_session():
     global IG_SESSION, IG_SESSION_EXPIRY
@@ -347,26 +179,7 @@ def parse_cookie_string(cookie_str):
     return cookies
-def validate_imported_session(session):
-    """Check if the imported session seems valid"""
-    # Check for essential cookies
-    has_sessionid = any(c['name'] == 'sessionid' for c in session.cookies._cookies.get('.instagram.com', {}).get('/', {}).values())
-    if not has_sessionid:
-        return False
-    # Try to make a basic request
-    try:
-        response = session.get('https://www.instagram.com/accounts/edit/', allow_redirects=False)
-        # If we get redirected to login page, session is invalid
-        if response.status_code == 302 and 'login' in response.headers.get('Location', ''):
-            return False
-        return response.status_code == 200
-    except:
-        return False
 def get_instagram_stories(username):
     session = get_instagram_session()
     if not session:
@@ -380,7 +193,6 @@ def get_instagram_stories(username):
         session.headers.update({
             'User-Agent': random.choice(USER_AGENTS),
             'X-IG-App-ID': '936619743392459',
-            'X-IG-WWW-Claim': 'hmac.AR1k18eNmvbsN7COEa9jpSQ9-NKGlK_TVqhgqxaGgxdCsIfR',
             'Referer': f'https://www.instagram.com/{username}/',
         })
@@ -458,19 +270,75 @@ def get_instagram_stories(username):
         logger.error(f"Error getting stories: {str(e)}")
         return {"error": "exception", "message": str(e)}
-# Web routes
 @app.get("/", response_class=HTMLResponse)
 async def root():
     return RedirectResponse(url="/status")
 @app.get("/import", response_class=HTMLResponse)
 async def import_page(message: str = None, success: bool = False):
-    status_html = ""
-    if message:
-        status_class = "success" if success else "error"
-        status_html = f'<div class="status {status_class}">{message}</div>'
-    return HTMLResponse(COOKIE_IMPORT_HTML % status_html)
 @app.post("/import-cookies")
 async def import_cookies(cookies: str = Form(...), username: str = Form(...)):
@@ -481,134 +349,109 @@ async def import_cookies(cookies: str = Form(...), username: str = Form(...)):
         if not cookie_objects:
             logger.warning("No valid cookies found in input")
-            return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">No valid cookies found in input</div>')
         # Check for essential cookies
         has_sessionid = any(c['name'] == 'sessionid' for c in cookie_objects)
         if not has_sessionid:
             logger.warning("Missing essential sessionid cookie")
-            return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Missing essential sessionid cookie</div>')
         # Save session
         if save_instagram_session(cookie_objects, username):
             # Verify session
             session = get_instagram_session()
             if session:
-                # Try to make a basic request to test session
-                try:
-                    response = session.get('https://www.instagram.com/accounts/edit/', allow_redirects=False)
-                    if response.status_code == 302 and 'login' in response.headers.get('Location', ''):
-                        logger.warning("Session seems invalid - redirected to login")
-                        clear_instagram_session()
-                        return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Session validation failed - redirected to login</div>')
-                    logger.info("Session validated successfully")
-                    return RedirectResponse(url="/status", status_code=303)
-                except Exception as e:
-                    logger.error(f"Session validation error: {str(e)}")
-                    return HTMLResponse(COOKIE_IMPORT_HTML % f'<div class="status error">Session validation error: {str(e)}</div>')
             else:
                 logger.error("Failed to get session after saving")
-                return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Failed to create session</div>')
         else:
             logger.error("Failed to save session")
-            return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Failed to save session</div>')
     except Exception as e:
         logger.error(f"Import error: {str(e)}")
-        return HTMLResponse(COOKIE_IMPORT_HTML % f'<div class="status error">Import error: {str(e)}</div>')
-@app.post("/import-cookies-json")
-async def import_cookies_json(cookies_json: str = Form(...), username: str = Form(...)):
-    logger.info(f"Importing JSON cookies for {username}")
-    try:
-        cookie_objects = json.loads(cookies_json)
-        if not isinstance(cookie_objects, list):
-            logger.warning("Invalid JSON format - expected array")
-            return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Invalid JSON format - expected array of cookie objects</div>')
-        # Check for essential cookies
-        has_sessionid = any(c.get('name') == 'sessionid' for c in cookie_objects)
-        if not has_sessionid:
-            logger.warning("Missing essential sessionid cookie")
-            return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Missing essential sessionid cookie</div>')
-        # Normalize cookie objects
-        for cookie in cookie_objects:
-            if 'domain' not in cookie:
-                cookie['domain'] = '.instagram.com'
-        # Save session
-        if save_instagram_session(cookie_objects, username):
-            # Verify session
-            session = get_instagram_session()
-            if session:
-                # Try to make a basic request to test session
-                try:
-                    response = session.get('https://www.instagram.com/accounts/edit/', allow_redirects=False)
-                    if response.status_code == 302 and 'login' in response.headers.get('Location', ''):
-                        logger.warning("Session seems invalid - redirected to login")
-                        clear_instagram_session()
-                        return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Session validation failed - redirected to login</div>')
-                    logger.info("Session validated successfully")
-                    return RedirectResponse(url="/status", status_code=303)
-                except Exception as e:
-                    logger.error(f"Session validation error: {str(e)}")
-                    return HTMLResponse(COOKIE_IMPORT_HTML % f'<div class="status error">Session validation error: {str(e)}</div>')
-            else:
-                logger.error("Failed to get session after saving")
-                return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Failed to create session</div>')
-        else:
-            logger.error("Failed to save session")
-            return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Failed to save session</div>')
-    except json.JSONDecodeError:
-        logger.error("Invalid JSON format")
-        return HTMLResponse(COOKIE_IMPORT_HTML % '<div class="status error">Invalid JSON format</div>')
-    except Exception as e:
-        logger.error(f"Import error: {str(e)}")
-        return HTMLResponse(COOKIE_IMPORT_HTML % f'<div class="status error">Import error: {str(e)}</div>')
-@app.get("/logout")
-async def logout():
-    clear_instagram_session()
-    return RedirectResponse(url="/status", status_code=303)
 @app.get("/status", response_class=HTMLResponse)
 async def status_page():
     session = get_instagram_session()
     session_valid = session is not None
-    # Status badge
-    status_badge = '<span class="status connected">Connected</span>' if session_valid else '<span class="status disconnected">Disconnected</span>'
-    # Status info
     if session_valid:
-        status_info = f"<p>✅ You are currently logged in as <strong>{IG_SESSION_USERNAME}</strong></p>"
-        status_info += f"<p>Session will expire: {IG_SESSION_EXPIRY.strftime('%Y-%m-%d %H:%M:%S')}</p>"
     else:
-        status_info = "<p>❌ No active Instagram session</p><p>Please import your cookies to use the API</p>"
-    # Button text
-    import_text = "Re-Import Cookies" if session_valid else "Import Cookies"
-    # Logout button
-    logout_button = '<a href="/logout" class="btn btn-danger">Logout</a>' if session_valid else ''
-    # Cache info
-    cache_count = len(STORY_CACHE)
-    story_count = sum(len(data.get("data", [])) for data in STORY_CACHE.values())
-    return HTMLResponse(STATUS_HTML % (status_badge, status_info, import_text, logout_button, cache_count, story_count))
-# API routes
 @app.get("/stories/{username}")
 async def get_stories(username: str, cached: bool = False):
     logger.info(f"Request for @{username} stories")
@@ -727,6 +570,19 @@ async def download_media(url: str):
             detail="Failed to download media"
         )
 # Load cache and session at startup
 @app.on_event("startup")
 def startup_event():
@@ -754,17 +610,4 @@ def startup_event():
         logger.error(f"Cache loading error: {e}")
     # Try to load session
-    get_instagram_session()
-# Health check endpoint
-@app.get("/health")
-async def health_check():
-    session = get_instagram_session()
-    return {
-        "status": "ok",
-        "authenticated": session is not None,
-        "username": IG_SESSION_USERNAME,
-        "session_expires": IG_SESSION_EXPIRY.isoformat() if IG_SESSION_EXPIRY else None,
-        "cache_items": len(STORY_CACHE),
-        "timestamp": datetime.now().isoformat()
-    }

+from fastapi import FastAPI, HTTPException, status, Form
 from fastapi.responses import StreamingResponse, HTMLResponse, RedirectResponse
 import requests
 import os
 import random
 import json
 from datetime import datetime, timedelta
 # Configure logging
 logging.basicConfig(
     "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
 ]
+# Functions for cache management
 def get_cache_key(username):
     return username.lower()
     LAST_REQUEST[key] = now
     return False
+# Session management functions
 def get_instagram_session():
     global IG_SESSION, IG_SESSION_EXPIRY
     return cookies
+# Instagram API functions
 def get_instagram_stories(username):
     session = get_instagram_session()
     if not session:
         session.headers.update({
             'User-Agent': random.choice(USER_AGENTS),
             'X-IG-App-ID': '936619743392459',
             'Referer': f'https://www.instagram.com/{username}/',
         })
         logger.error(f"Error getting stories: {str(e)}")
         return {"error": "exception", "message": str(e)}
+# Routes
 @app.get("/", response_class=HTMLResponse)
 async def root():
     return RedirectResponse(url="/status")
 @app.get("/import", response_class=HTMLResponse)
 async def import_page(message: str = None, success: bool = False):
+    status_class = "success" if success else "error"
+    status_div = f'<div class="status {status_class}">{message}</div>' if message else ""
+    html = """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>Import Instagram Session</title>
+        <style>
+            body { font-family: Arial; max-width: 700px; margin: 0 auto; padding: 20px; }
+            .container { background-color: #fff; border-radius: 8px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); padding: 25px; }
+            h1 { color: #262626; text-align: center; }
+            .form-group { margin-bottom: 15px; }
+            label { display: block; margin-bottom: 10px; }
+            textarea { width: 100%; padding: 10px; border: 1px solid #dbdbdb; border-radius: 3px; height: 150px; }
+            input[type="text"] { width: 100%; padding: 10px; border: 1px solid #dbdbdb; border-radius: 3px; }
+            .btn { background-color: #0095f6; border: none; color: white; padding: 10px 15px; border-radius: 4px; cursor: pointer; font-weight: bold; width: 100%; }
+            .status { margin-top: 20px; padding: 10px; border-radius: 4px; }
+            .success { background-color: #e8f5e9; color: #388e3c; }
+            .error { background-color: #ffebee; color: #d32f2f; }
+            .instructions { background-color: #f8f9fa; padding: 15px; border-radius: 4px; margin-bottom: 20px; }
+            .code { font-family: monospace; background-color: #f1f1f1; padding: 2px 5px; border-radius: 3px; }
+            ol { padding-left: 20px; }
+            li { margin-bottom: 10px; }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>Import Instagram Session</h1>
+            <div class="instructions">
+                <h3>How to get your Instagram cookies:</h3>
+                <ol>
+                    <li>Log in to Instagram in your browser</li>
+                    <li>Open browser DevTools (F12 or right-click and select "Inspect")</li>
+                    <li>Go to the "Application" or "Storage" tab</li>
+                    <li>Under "Cookies", find "instagram.com"</li>
+                    <li>Find the <span class="code">sessionid</span> cookie and copy its value</li>
+                    <li>Also copy the <span class="code">ds_user_id</span> and <span class="code">csrftoken</span> cookies if available</li>
+                    <li>Paste the cookies in the format: <span class="code">name=value; name2=value2;</span></li>
+                </ol>
+            </div>
+            """ + status_div + """
+            <form method="post" action="/import-cookies">
+                <div class="form-group">
+                    <label for="cookies">Paste your Instagram cookies here:</label>
+                    <textarea id="cookies" name="cookies" placeholder="sessionid=YOUR_SESSION_ID; ds_user_id=YOUR_USER_ID; csrftoken=YOUR_CSRF_TOKEN;" required></textarea>
+                </div>
+                <div class="form-group">
+                    <label for="username">Your Instagram username:</label>
+                    <input type="text" id="username" name="username" required>
+                </div>
+                <button type="submit" class="btn">Import Session</button>
+            </form>
+        </div>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html)
 @app.post("/import-cookies")
 async def import_cookies(cookies: str = Form(...), username: str = Form(...)):
         if not cookie_objects:
             logger.warning("No valid cookies found in input")
+            return import_page(message="No valid cookies found in input")
         # Check for essential cookies
         has_sessionid = any(c['name'] == 'sessionid' for c in cookie_objects)
         if not has_sessionid:
             logger.warning("Missing essential sessionid cookie")
+            return import_page(message="Missing essential sessionid cookie")
         # Save session
         if save_instagram_session(cookie_objects, username):
             # Verify session
             session = get_instagram_session()
             if session:
+                logger.info("Session imported successfully")
+                return RedirectResponse(url="/status", status_code=303)
             else:
                 logger.error("Failed to get session after saving")
+                return import_page(message="Failed to create session")
         else:
             logger.error("Failed to save session")
+            return import_page(message="Failed to save session")
     except Exception as e:
         logger.error(f"Import error: {str(e)}")
+        return import_page(message=f"Import error: {str(e)}")
 @app.get("/status", response_class=HTMLResponse)
 async def status_page():
     session = get_instagram_session()
     session_valid = session is not None
+    cache_count = len(STORY_CACHE)
+    story_count = sum(len(data.get("data", [])) for data in STORY_CACHE.values())
+    status_badge = 'Connected' if session_valid else 'Disconnected'
+    status_class = 'connected' if session_valid else 'disconnected'
     if session_valid:
+        status_info = f"✅ Logged in as <strong>{IG_SESSION_USERNAME}</strong><br>"
+        status_info += f"Session expires: {IG_SESSION_EXPIRY.strftime('%Y-%m-%d %H:%M:%S')}"
     else:
+        status_info = "❌ No active Instagram session<br>Please import your cookies to use the API"
+    html = f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>API Status</title>
+        <style>
+            body {{ font-family: Arial; max-width: 600px; margin: 0 auto; padding: 20px; }}
+            .container {{ background-color: #fff; border-radius: 8px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); padding: 25px; }}
+            h1 {{ color: #262626; text-align: center; }}
+            .status-badge {{ display: inline-block; padding: 5px 10px; border-radius: 4px; font-weight: bold; margin-left: 10px; }}
+            .connected {{ background-color: #e8f5e9; color: #388e3c; }}
+            .disconnected {{ background-color: #ffebee; color: #d32f2f; }}
+            .info {{ background-color: #f5f5f5; border-radius: 4px; padding: 15px; margin: 15px 0; }}
+            .btn {{ background-color: #0095f6; border: none; color: white; padding: 10px 15px; border-radius: 4px; text-decoration: none; display: inline-block; margin-right: 10px; }}
+            .btn-danger {{ background-color: #f44336; }}
+            table {{ width: 100%; border-collapse: collapse; margin-top: 15px; }}
+            th, td {{ text-align: left; padding: 8px; border-bottom: 1px solid #ddd; }}
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>Instagram API Status</h1>
+            <h2>Connection Status
+                <span class="status-badge {status_class}">{status_badge}</span>
+            </h2>
+            <div class="info">
+                {status_info}
+            </div>
+            <a href="/import" class="btn">{"Re-Import Cookies" if session_valid else "Import Cookies"}</a>
+            {"<a href='/logout' class='btn btn-danger'>Logout</a>" if session_valid else ""}
+            <h2>API Endpoints</h2>
+            <table>
+                <tr><th>Endpoint</th><th>Description</th></tr>
+                <tr><td><code>/stories/{{username}}</code></td><td>Get stories for a user</td></tr>
+                <tr><td><code>/download/{{url}}</code></td><td>Download media from a URL</td></tr>
+                <tr><td><code>/status</code></td><td>View API status</td></tr>
+            </table>
+            <h2>Cache Info</h2>
+            <div class="info">
+                <p>Cached users: {cache_count}</p>
+                <p>Total stories cached: {story_count}</p>
+            </div>
+        </div>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html)
+@app.get("/logout")
+async def logout():
+    clear_instagram_session()
+    return RedirectResponse(url="/status", status_code=303)
 @app.get("/stories/{username}")
 async def get_stories(username: str, cached: bool = False):
     logger.info(f"Request for @{username} stories")
             detail="Failed to download media"
         )
+# Health check endpoint
+@app.get("/health")
+async def health_check():
+    session = get_instagram_session()
+    return {
+        "status": "ok",
+        "authenticated": session is not None,
+        "username": IG_SESSION_USERNAME,
+        "session_expires": IG_SESSION_EXPIRY.isoformat() if IG_SESSION_EXPIRY else None,
+        "cache_items": len(STORY_CACHE),
+        "timestamp": datetime.now().isoformat()
+    }
 # Load cache and session at startup
 @app.on_event("startup")
 def startup_event():
         logger.error(f"Cache loading error: {e}")
     # Try to load session
+    get_instagram_session()