Update app.py

app.py

@@ -1,10 +1,11 @@
 from fastapi import FastAPI, File, UploadFile, Request
-from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
+from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse, FileResponse
 import requests
 import time
 import asyncio
 from typing import Dict
 import os
+import mimetypes
 
 app = FastAPI()
 
@@ -424,7 +425,7 @@ HTML_CONTENT = """
             gap: 5px;
         }
 
-                .quick-open-modal {
+        .quick-open-modal {
             display: none;
             position: fixed;
             z-index: 4;
@@ -774,11 +775,11 @@ HTML_CONTENT = """
             linkContainer.style.marginBottom = '10px';
 
             const link = document.createElement('a');
-            link.href = url;
+            link.href = url.replace('.png', '');
             link.textContent = `View ${fileName}`;
             link.className = 'result-link';
             link.target = '_blank';
-            link.download = `${fileName.split('.')[0]}.${originalExtension}`;
+            link.download = fileName;
 
             linkContainer.appendChild(link);
 
@@ -789,7 +790,7 @@ HTML_CONTENT = """
             copyBtn.textContent = 'Copy Link';
             copyBtn.className = 'small-btn copy-btn';
             copyBtn.onclick = () => {
-                navigator.clipboard.writeText(window.location.origin + url).then(() => {
+                navigator.clipboard.writeText(window.location.origin + url.replace('.png', '')).then(() => {
                     alert('Link copied to clipboard!');
                 });
             };
@@ -800,7 +801,7 @@ HTML_CONTENT = """
                 embedBtn.textContent = 'Embed Video for Discord';
                 embedBtn.className = 'small-btn embed-btn';
                 embedBtn.onclick = () => {
-                    showEmbedModal(url);
+                    showEmbedModal(url.replace('.png', ''));
                 };
                 buttonsContainer.appendChild(embedBtn);
             }
@@ -825,7 +826,7 @@ HTML_CONTENT = """
 
         function saveToHistory(fileName, url, originalExtension) {
             let history = JSON.parse(localStorage.getItem('uploadHistory')) || [];
-            history.unshift({ fileName, url, originalExtension, timestamp: new Date().toISOString() });
+            history.unshift({ fileName, url: url.replace('.png', ''), originalExtension, timestamp: new Date().toISOString() });
             if (history.length > 500) history = history.slice(0, 500);
             localStorage.setItem('uploadHistory', JSON.stringify(history));
         }
@@ -924,7 +925,7 @@ HTML_CONTENT = """
                 link.href = fullUrl;
                 link.textContent = 'Download ' + fileName;
                 link.target = '_blank';
-                link.download = `${fileName.split('.')[0]}.${originalExtension}`;
+                link.download = fileName;
                 quickOpenContent.appendChild(link);
             }
 
@@ -949,14 +950,21 @@ async def handle_upload(file: UploadFile = File(...)):
         return JSONResponse(content={"error": "Failed"}, status_code=500)
 
     original_extension = os.path.splitext(file.filename)[1][1:]
-    temp_filename = f"{file.filename}.png"
-
-    upload_result = await initiate_upload(cookies, temp_filename, "image/png")
+    supported_types = ['mp4', 'png', 'zip', 'pdf', 'txt']
+    
+    if original_extension.lower() in supported_types:
+        temp_filename = file.filename
+        content_type = file.content_type
+    else:
+        temp_filename = f"{file.filename}.png"
+        content_type = "image/png"
+
+    upload_result = await initiate_upload(cookies, temp_filename, content_type)
     if not upload_result or 'upload_url' not in upload_result:
         return JSONResponse(content={"error": "Failed to upload"}, status_code=500)
 
     file_content = await file.read()
-    upload_success = await retry_upload(upload_result['upload_url'], file_content, "image/png")
+    upload_success = await retry_upload(upload_result['upload_url'], file_content, content_type)
     if not upload_success:
         return JSONResponse(content={"error": "FAILED GOD MAN AFTER alot of attempts"}, status_code=500)
 
@@ -984,6 +992,13 @@ async def handle_file_stream(path: str, request: Request):
     if response.status_code == 206:
         headers['Content-Range'] = response.headers.get('Content-Range')
 
+    # Remove .png extension if present
+    if path.lower().endswith('.png'):
+        original_filename = os.path.splitext(path)[0]
+        content_type, _ = mimetypes.guess_type(original_filename)
+        if content_type:
+            headers['Content-Type'] = content_type
+
     return StreamingResponse(generate(), status_code=response.status_code, headers=headers)
 
 @app.get("/embed")