Update app.py

app.py

@@ -9,25 +9,12 @@ import mimetypes
 import httpx
 import io
 import zipfile
+import math
 
 app = FastAPI()
 
 DENO_API_URL = "https://dataerrr99.deno.dev"
 
-# Define allowed media types
-ALLOWED_MEDIA_TYPES = {
-    "image": ["image/jpeg", "image/png", "image/gif", "image/webp", "image/svg+xml"],
-    "video": ["video/mp4", "video/mkv", "video/webm", "video/quicktime"],
-    "audio": ["audio/mpeg", "audio/wav", "audio/ogg", "audio/webm"]
-}
-
-def is_media_file(content_type: str) -> bool:
-    """Check if the file is a media type (image, video, or audio)."""
-    for category, types in ALLOWED_MEDIA_TYPES.items():
-        if content_type in types:
-            return True
-    return False
-
 async def save_album_to_db(album_name: str, album_id: str, files: List[dict], is_private: bool = False):
     async with httpx.AsyncClient() as client:
         try:
@@ -68,8 +55,6 @@ def get_file_type(filename):
             return 'image'
         elif mime_type.startswith('video/'):
             return 'video'
-        elif mime_type.startswith('audio/'):
-            return 'audio'
     return 'other'
 
 @app.get("/", response_class=HTMLResponse)
@@ -171,10 +156,6 @@ async def create_album(
     uploaded_files = 0
     
     for file in files:
-        if not is_media_file(file.content_type):
-            # Redirect to single file upload for non-media files
-            return RedirectResponse(url="https://albumup-up1.hf.space/", status_code=303)
-        
         file_content = await file.read()
         upload_result = await initiate_upload(cookies, file.filename, file.content_type)
         
@@ -226,13 +207,10 @@ async def view_album(album_id: str):
                     </video>
                 </div>
             """
-        elif file_type == 'audio':
+        else:
             preview_html = f"""
                 <div class="preview-container">
-                    <audio controls>
-                        <source src="{file_url}" type="{file['content_type']}">
-                        Your browser does not support the audio tag.
-                    </audio>
+                    <p>No preview available for {file['filename']}</p>
                 </div>
             """
 
@@ -376,4 +354,4 @@ async def retry_upload(upload_url, file_content, content_type, max_retries=5):
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)