Update app.py

app.py

@@ -9,9 +9,6 @@ import mimetypes
 import httpx
 import io
 import zipfile
-import math
-from PIL import Image
-import os
 
 app = FastAPI()
 
@@ -174,22 +171,11 @@ async def create_album(
     uploaded_files = 0
     
     for file in files:
-        file_content = await file.read()
-        original_extension = os.path.splitext(file.filename)[1].lower()
-        
-        # Convert non-media files to PNG
         if not is_media_file(file.content_type):
-            try:
-                # Create a blank image with the file content as text
-                img = Image.new('RGB', (200, 200), color=(73, 109, 137))
-                img_byte_arr = io.BytesIO()
-                img.save(img_byte_arr, format='PNG')
-                file_content = img_byte_arr.getvalue()
-                file.content_type = "image/png"
-            except Exception as e:
-                print(f"Error converting file to PNG: {str(e)}")
-                continue
+            # 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)
         
         if upload_result and 'upload_url' in upload_result:
@@ -200,7 +186,6 @@ async def create_album(
                     'filename': file.filename,
                     'path': serving_path,
                     'content_type': file.content_type,
-                    'original_extension': original_extension,  # Save original extension
                     'uploaded_at': datetime.now().isoformat()
                 })
                 uploaded_files += 1
@@ -250,12 +235,6 @@ async def view_album(album_id: str):
                     </audio>
                 </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">
@@ -294,11 +273,7 @@ async def download_album(album_id: str):
     with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
         for file in album['files']:
             response = requests.get(f"https://replicate.delivery/pbxt/{file['path']}")
-            # Restore the original file extension
-            original_filename = file['filename']
-            if 'original_extension' in file:
-                original_filename = os.path.splitext(file['filename'])[0] + file['original_extension']
-            zip_file.writestr(original_filename, response.content)
+            zip_file.writestr(file['filename'], response.content)
     
     zip_buffer.seek(0)