Spaces:

albumup
/

file

Sleeping

App Files Files Community

albumup commited on Jan 17, 2025

Commit

1982de9

verified ·

1 Parent(s): c9d20aa

Update app.py

Browse files

Files changed (1) hide show

app.py +380 -49

app.py CHANGED Viewed

@@ -10,6 +10,7 @@ import uuid
 import json
 from datetime import datetime
 import httpx
 from functools import lru_cache
 class Settings(BaseSettings):
@@ -30,34 +31,8 @@ class Album(BaseModel):
 app = FastAPI()
-# In-memory cache for faster access
 albums_cache = {}
-async def sync_with_database():
-    settings = get_settings()
-    try:
-        async with httpx.AsyncClient() as client:
-            response = await client.get(f"{settings.database_url}/albums")
-            if response.status_code == 200:
-                data = response.json()
-                albums_cache.clear()
-                albums_cache.update(data)
-    except Exception as e:
-        print(f"Database sync error: {e}")
-async def save_to_database(album_id: str, album_data: dict):
-    settings = get_settings()
-    try:
-        async with httpx.AsyncClient() as client:
-            response = await client.post(
-                f"{settings.database_url}/albums",
-                json={album_id: album_data}
-            )
-            if response.status_code != 200:
-                raise HTTPException(status_code=500, detail="Failed to save to database")
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
 HTML_CONTENT = """
 <!DOCTYPE html>
 <html lang="en">
@@ -77,28 +52,50 @@ HTML_CONTENT = """
             border: 1px solid #ddd;
             border-radius: 5px;
         }
         .search-form {
             margin: 20px 0;
             padding: 20px;
             border: 1px solid #ddd;
             border-radius: 5px;
         }
     </style>
 </head>
 <body>
     <div class="container">
         <h1>File Upload & Album Management</h1>
-        <!-- Search Albums -->
         <div class="search-form">
             <h2>Search Albums</h2>
             <form action="/search" method="get">
                 <input type="text" name="query" placeholder="Search by album name..." required>
-                <button type="submit">Search</button>
             </form>
         </div>
-        <!-- Album Upload -->
         <div class="upload-form">
             <h2>Create Album</h2>
             <form action="/album/create" method="post" enctype="multipart/form-data">
@@ -106,13 +103,271 @@ HTML_CONTENT = """
                 <input type="checkbox" name="is_private" id="is_private">
                 <label for="is_private">Make Album Private</label><br><br>
                 <input type="file" name="files" accept="*/*" multiple required><br><br>
-                <button type="submit">Create Album</button>
             </form>
         </div>
     </div>
 </body>
 </html>
 """
 @app.on_event("startup")
 async def startup_event():
@@ -122,15 +377,35 @@ async def startup_event():
 async def index():
     return HTML_CONTENT
-@app.get("/search")
-async def search_albums(query: str = "", include_private: bool = False):
     matching_albums = {
         album_id: album for album_id, album in albums_cache.items()
-        if (query.lower() in album['name'].lower() and
-            (not album['is_private'] or include_private))
     }
-    return {"albums": matching_albums}
 @app.post("/album/create")
 async def create_album(
     request: Request,
@@ -163,7 +438,7 @@ async def create_album(
         'created_at': datetime.now().isoformat(),
         'is_private': is_private,
         'files': album_files,
-        'owner': 'anonymous'  # You can implement user authentication later
     }
     albums_cache[album_id] = album_data
@@ -172,17 +447,60 @@ async def create_album(
     base_url = str(request.base_url).rstrip('/')
     return RedirectResponse(url=f"{base_url}/album/{album_id}", status_code=303)
-@app.get("/album/{album_id}")
-async def view_album(album_id: str, request: Request):
     if album_id not in albums_cache:
-        raise HTTPException(status_code=404, detail="Album not found")
     album = albums_cache[album_id]
-    if album['is_private']:
-        # Implement authentication check here
-        pass
-    return album
 @app.get("/album/{album_id}/download")
 async def download_album(album_id: str):
@@ -190,10 +508,7 @@ async def download_album(album_id: str):
         raise HTTPException(status_code=404, detail="Album not found")
     album = albums_cache[album_id]
-    if album['is_private']:
-        # Implement authentication check here
-        pass
     import io
     import zipfile
@@ -213,7 +528,23 @@ async def download_album(album_id: str):
         }
     )
-# Helper functions remain the same as in previous version
 async def get_cookies():
     try:
         response = requests.get('https://replicate.com/levelsio/neon-tokyo', headers={

 import json
 from datetime import datetime
 import httpx
+import mimetypes
 from functools import lru_cache
 class Settings(BaseSettings):
 app = FastAPI()
 albums_cache = {}
 HTML_CONTENT = """
 <!DOCTYPE html>
 <html lang="en">
             border: 1px solid #ddd;
             border-radius: 5px;
         }
+        .preview {
+            margin-top: 20px;
+        }
         .search-form {
             margin: 20px 0;
             padding: 20px;
             border: 1px solid #ddd;
             border-radius: 5px;
         }
+        .album-list {
+            margin-top: 20px;
+        }
+        .album-item {
+            padding: 10px;
+            border: 1px solid #ddd;
+            margin-bottom: 10px;
+            border-radius: 5px;
+        }
+        .button {
+            display: inline-block;
+            padding: 8px 16px;
+            background-color: #007bff;
+            color: white;
+            text-decoration: none;
+            border-radius: 4px;
+            margin: 5px;
+        }
+        .button:hover {
+            background-color: #0056b3;
+        }
     </style>
 </head>
 <body>
     <div class="container">
         <h1>File Upload & Album Management</h1>
         <div class="search-form">
             <h2>Search Albums</h2>
             <form action="/search" method="get">
                 <input type="text" name="query" placeholder="Search by album name..." required>
+                <button type="submit" class="button">Search</button>
             </form>
         </div>
         <div class="upload-form">
             <h2>Create Album</h2>
             <form action="/album/create" method="post" enctype="multipart/form-data">
                 <input type="checkbox" name="is_private" id="is_private">
                 <label for="is_private">Make Album Private</label><br><br>
                 <input type="file" name="files" accept="*/*" multiple required><br><br>
+                <button type="submit" class="button">Create Album</button>
             </form>
         </div>
     </div>
 </body>
 </html>
 """
+ALBUM_VIEW_HTML = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Album View - {album_name}</title>
+    <style>
+        .container {{
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+        }}
+        .file-grid {{
+            display: grid;
+            grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
+            gap: 20px;
+        }}
+        .file-item {{
+            border: 1px solid #ddd;
+            padding: 10px;
+            text-align: center;
+            border-radius: 5px;
+        }}
+        .download-all {{
+            margin: 20px 0;
+        }}
+        .back-link {{
+            margin-bottom: 20px;
+            display: block;
+        }}
+        .button {{
+            display: inline-block;
+            padding: 8px 16px;
+            background-color: #007bff;
+            color: white;
+            text-decoration: none;
+            border-radius: 4px;
+            margin: 5px;
+        }}
+        .button:hover {{
+            background-color: #0056b3;
+        }}
+        .preview-container {{
+            margin-bottom: 10px;
+            max-width: 100%;
+            height: 200px;
+            overflow: hidden;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            background-color: #f8f9fa;
+            border-radius: 4px;
+        }}
+        .preview-container img {{
+            max-width: 100%;
+            max-height: 100%;
+            object-fit: contain;
+        }}
+        .preview-container video {{
+            max-width: 100%;
+            max-height: 100%;
+            object-fit: contain;
+        }}
+        .modal {{
+            display: none;
+            position: fixed;
+            z-index: 1000;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+            background-color: rgba(0,0,0,0.9);
+            overflow: auto;
+        }}
+        .modal-content {{
+            margin: auto;
+            display: block;
+            max-width: 90%;
+            max-height: 90vh;
+            margin-top: 50px;
+        }}
+        .close {{
+            position: absolute;
+            right: 35px;
+            top: 15px;
+            color: #f1f1f1;
+            font-size: 40px;
+            font-weight: bold;
+            cursor: pointer;
+        }}
+        .privacy-badge {{
+            display: inline-block;
+            padding: 5px 10px;
+            border-radius: 4px;
+            margin: 10px 0;
+            color: white;
+        }}
+        .private {{
+            background-color: #dc3545;
+        }}
+        .public {{
+            background-color: #28a745;
+        }}
+    </style>
+    <script>
+        function openModal(url) {{
+            const modal = document.getElementById('previewModal');
+            const modalContent = document.getElementById('modalContent');
+            const contentType = getContentType(url);
+            if (contentType === 'image') {{
+                modalContent.innerHTML = `<img src="${{url}}" style="max-width:100%;max-height:90vh;">`;
+            }} else if (contentType === 'video') {{
+                modalContent.innerHTML = `<video controls style="max-width:100%;max-height:90vh;">
+                    <source src="${{url}}" type="video/mp4">
+                    Your browser does not support the video tag.
+                </video>`;
+            }}
+            modal.style.display = 'block';
+        }}
+        function closeModal() {{
+            const modal = document.getElementById('previewModal');
+            modal.style.display = 'none';
+            const modalContent = document.getElementById('modalContent');
+            modalContent.innerHTML = '';
+        }}
+        function getContentType(url) {{
+            const ext = url.split('.').pop().toLowerCase();
+            if (['jpg', 'jpeg', 'png', 'gif', 'webp'].includes(ext)) return 'image';
+            if (['mp4', 'webm', 'ogg'].includes(ext)) return 'video';
+            return 'other';
+        }}
+    </script>
+</head>
+<body>
+    <div class="container">
+        <a href="/" class="back-link button">← Back to Home</a>
+        <h1>{album_name}</h1>
+        <div class="privacy-badge {privacy_class}">{privacy_status}</div>
+        <div class="download-all">
+            <a href="/album/{album_id}/download" class="button" target="_blank">Download All Files</a>
+        </div>
+        <div class="file-grid">
+            {file_list}
+        </div>
+    </div>
+    <div id="previewModal" class="modal" onclick="closeModal()">
+        <span class="close">&times;</span>
+        <div id="modalContent" class="modal-content">
+        </div>
+    </div>
+</body>
+</html>
+"""
+SEARCH_RESULTS_HTML = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Search Results</title>
+    <style>
+        .container {{
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+        }}
+        .album-item {{
+            border: 1px solid #ddd;
+            padding: 15px;
+            margin-bottom: 15px;
+            border-radius: 5px;
+        }}
+        .back-link {{
+            margin-bottom: 20px;
+            display: block;
+        }}
+        .button {{
+            display: inline-block;
+            padding: 8px 16px;
+            background-color: #007bff;
+            color: white;
+            text-decoration: none;
+            border-radius: 4px;
+            margin: 5px;
+        }}
+        .button:hover {{
+            background-color: #0056b3;
+        }}
+        .privacy-badge {{
+            display: inline-block;
+            padding: 5px 10px;
+            border-radius: 4px;
+            margin-left: 10px;
+            color: white;
+        }}
+        .private {{
+            background-color: #dc3545;
+        }}
+        .public {{
+            background-color: #28a745;
+        }}
+    </style>
+</head>
+<body>
+    <div class="container">
+        <a href="/" class="back-link button">← Back to Home</a>
+        <h1>Search Results</h1>
+        <p>Found {result_count} albums for "{query}"</p>
+        <div class="search-results">
+            {results}
+        </div>
+    </div>
+</body>
+</html>
+"""
+async def sync_with_database():
+    settings = get_settings()
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.get(f"{settings.database_url}/albums")
+            if response.status_code == 200:
+                data = response.json()
+                albums_cache.clear()
+                albums_cache.update(data)
+    except Exception as e:
+        print(f"Database sync error: {e}")
+async def save_to_database(album_id: str, album_data: dict):
+    settings = get_settings()
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                f"{settings.database_url}",
+                json={
+                    "key": album_id,
+                    "value": album_data
+                }
+            )
+            if response.status_code != 200:
+                raise HTTPException(status_code=500, detail="Failed to save to database")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+def get_file_type(filename):
+    mime_type, _ = mimetypes.guess_type(filename)
+    if mime_type:
+        if mime_type.startswith('image/'):
+            return 'image'
+        elif mime_type.startswith('video/'):
+            return 'video'
+    return 'other'
 @app.on_event("startup")
 async def startup_event():
 async def index():
     return HTML_CONTENT
+@app.get("/search", response_class=HTMLResponse)
+async def search_albums(query: str = ""):
+    query = query.lower()
     matching_albums = {
         album_id: album for album_id, album in albums_cache.items()
+        if query in album['name'].lower() and not album.get('is_private', False)
     }
+    results_html = ""
+    for album_id, album in matching_albums.items():
+        privacy_status = "Private" if album.get('is_private') else "Public"
+        privacy_class = "private" if album.get('is_private') else "public"
+        results_html += f"""
+        <div class="album-item">
+            <h3>{album['name']}
+                <span class="privacy-badge {privacy_class}">{privacy_status}</span>
+            </h3>
+            <p>Created: {album['created_at']}</p>
+            <p>Files: {len(album['files'])}</p>
+            <a href="/album/{album_id}" class="button">View Album</a>
+        </div>
+        """
+    return SEARCH_RESULTS_HTML.format(
+        query=query,
+        result_count=len(matching_albums),
+        results=results_html
+    )
 @app.post("/album/create")
 async def create_album(
     request: Request,
         'created_at': datetime.now().isoformat(),
         'is_private': is_private,
         'files': album_files,
+        'owner': 'anonymous'
     }
     albums_cache[album_id] = album_data
     base_url = str(request.base_url).rstrip('/')
     return RedirectResponse(url=f"{base_url}/album/{album_id}", status_code=303)
+@app.get("/album/{album_id}", response_class=HTMLResponse)
+async def view_album(album_id: str):
     if album_id not in albums_cache:
+        return "Album not found", 404
     album = albums_cache[album_id]
+    file_list_html = ""
+    for file in album['files']:
+        file_url = f"/upload/{file['path']}"
+        file_type = get_file_type(file['filename'])
+        preview_html = ""
+        if file_type == 'image':
+            preview_html = f"""
+                <div class="preview-container">
+                    <img src="{file_url}" alt="{file['filename']}" onclick="openModal('{file_url}')">
+                </div>
+            """
+        elif file_type == 'video':
+            preview_html = f"""
+                <div class="preview-container">
+                    <video onclick="openModal('{file_url}')" style="cursor: pointer;">
+                        <source src="{file_url}" type="{file['content_type']}">
+                        Your browser does not support the video tag.
+                    </video>
+                </div>
+            """
+        else:
+            preview_html = f"""
+                <div class="preview-container">
+                    <p>No preview available for {file['filename']}</p>
+                </div>
+            """
+        file_list_html += f"""
+        <div class="file-item">
+            {preview_html}
+            <p>{file['filename']}</p>
+            <a href="{file_url}" class="button" onclick="openModal('{file_url}'); return false;">View</a>
+            <a href="{file_url}?download=true" class="button" target="_blank">Download</a>
+        </div>
+        """
+    privacy_status = "Private" if album['is_private'] else "Public"
+    privacy_class = "private" if album['is_private'] else "public"
+    return ALBUM_VIEW_HTML.format(
+        album_name=album['name'],
+        album_id=album_id,
+        file_list=file_list_html,
+        privacy_status=privacy_status,
+        privacy_class=privacy_class
+    )
 @app.get("/album/{album_id}/download")
 async def download_album(album_id: str):
         raise HTTPException(status_code=404, detail="Album not found")
     album = albums_cache[album_id]
     import io
     import zipfile
         }
     )
+@app.get("/upload/{path:path}")
+async def handle_stream(path: str, request: Request):
+    original_url = f'https://replicate.delivery/pbxt/{path}'
+    range_header = request.headers.get('Range')
+    headers = {'Range': range_header} if range_header else {}
+    response = requests.get(original_url, headers=headers, stream=True)
+    def generate():
+        for chunk in response.iter_content(chunk_size=8192):
+            yield chunk
+    headers = dict(response.headers)
+    headers['Access-Control-Allow-Origin'] = '*'
+    return StreamingResponse(generate(), headers=headers, status_code=response.status_code)
 async def get_cookies():
     try:
         response = requests.get('https://replicate.com/levelsio/neon-tokyo', headers={