Update app.py

app.py

@@ -775,7 +775,7 @@ HTML_CONTENT = """
             linkContainer.style.marginBottom = '10px';
 
             const link = document.createElement('a');
-            link.href = url.replace('.png', '');
+            link.href = url;
             link.textContent = `View ${fileName}`;
             link.className = 'result-link';
             link.target = '_blank';
@@ -790,7 +790,7 @@ HTML_CONTENT = """
             copyBtn.textContent = 'Copy Link';
             copyBtn.className = 'small-btn copy-btn';
             copyBtn.onclick = () => {
-                navigator.clipboard.writeText(window.location.origin + url.replace('.png', '')).then(() => {
+                navigator.clipboard.writeText(window.location.origin + url).then(() => {
                     alert('Link copied to clipboard!');
                 });
             };
@@ -801,7 +801,7 @@ HTML_CONTENT = """
                 embedBtn.textContent = 'Embed Video for Discord';
                 embedBtn.className = 'small-btn embed-btn';
                 embedBtn.onclick = () => {
-                    showEmbedModal(url.replace('.png', ''));
+                    showEmbedModal(url);
                 };
                 buttonsContainer.appendChild(embedBtn);
             }
@@ -826,7 +826,7 @@ HTML_CONTENT = """
 
         function saveToHistory(fileName, url, originalExtension) {
             let history = JSON.parse(localStorage.getItem('uploadHistory')) || [];
-            history.unshift({ fileName, url: url.replace('.png', ''), originalExtension, timestamp: new Date().toISOString() });
+            history.unshift({ fileName, url, originalExtension, timestamp: new Date().toISOString() });
             if (history.length > 500) history = history.slice(0, 500);
             localStorage.setItem('uploadHistory', JSON.stringify(history));
         }
@@ -950,21 +950,14 @@ async def handle_upload(file: UploadFile = File(...)):
         return JSONResponse(content={"error": "Failed"}, status_code=500)
 
     original_extension = os.path.splitext(file.filename)[1][1:]
-    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)
+    temp_filename = f"{file.filename}.png"
+
+    upload_result = await initiate_upload(cookies, temp_filename, "image/png")
     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, content_type)
+    upload_success = await retry_upload(upload_result['upload_url'], file_content, "image/png")
     if not upload_success:
         return JSONResponse(content={"error": "FAILED GOD MAN AFTER alot of attempts"}, status_code=500)
 
@@ -992,12 +985,11 @@ 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
+    # Set the correct Content-Type based on the original file extension
+    original_extension = os.path.splitext(path)[1][1:]
+    content_type, _ = mimetypes.guess_type(f"file.{original_extension}")
+    if content_type:
+        headers['Content-Type'] = content_type
 
     return StreamingResponse(generate(), status_code=response.status_code, headers=headers)