Update app.py

app.py

@@ -7,11 +7,14 @@ from PIL import Image
 import io
 import cv2
 import numpy as np
+from fastapi import FastAPI, File, UploadFile, Form
+from fastapi.middleware.wsgi import WSGIMiddleware
+from starlette.responses import JSONResponse
 
 # Set page config
 st.set_page_config(page_title="UltraVideoSpace", layout="wide", initial_sidebar_state="collapsed")
 
-# Custom CSS with advanced styling
+# Custom CSS
 st.markdown("""
 <style>
     @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap');
@@ -93,58 +96,49 @@ st.markdown("""
     .upload-zone:hover {
         background-color: rgba(229, 9, 20, 0.1);
     }
-    .progress-bar {
-        height: 6px;
-        background-color: #2b2b2b;
-        border-radius: 3px;
-        margin-top: 12px;
-    }
-    .progress-bar-fill {
-        height: 100%;
-        background-color: #e50914;
-        border-radius: 3px;
-        transition: width 0.3s ease;
+    #file-upload-status {
+        margin-top: 20px;
+        padding: 10px;
+        border-radius: 5px;
+        background-color: rgba(0, 0, 0, 0.5);
     }
 </style>
 """, unsafe_allow_html=True)
 
-# Constants
-MAX_FILE_SIZE = 10 * 1024 * 1024 * 1024  # 10GB in bytes
+# FastAPI app
+app = FastAPI()
 
-def save_uploaded_file(uploaded_file):
-    try:
-        if uploaded_file.size > MAX_FILE_SIZE:
-            st.error(f"File size exceeds the 10GB limit. Please upload a smaller file.")
-            return False
-
-        save_path = Path("uploaded_videos")
-        save_path.mkdir(exist_ok=True)
-        file_path = save_path / uploaded_file.name
-        
-        total_size = uploaded_file.size
-        chunk_size = 1024 * 1024  # 1MB chunks
-        uploaded_size = 0
-
-        progress_bar = st.progress(0)
-        status_text = st.empty()
-
-        with file_path.open("wb") as f:
-            while True:
-                chunk = uploaded_file.read(chunk_size)
-                if not chunk:
-                    break
-                f.write(chunk)
-                uploaded_size += len(chunk)
-                progress = min(uploaded_size / total_size, 1.0)
-                progress_bar.progress(int(progress * 100))
-                status_text.text(f"Uploading... {int(progress * 100)}%")
-
-        progress_bar.empty()
-        status_text.empty()
-        return True
-    except Exception as e:
-        st.error(f"Error saving file: {str(e)}")
-        return False
+@app.post("/upload_chunk")
+async def upload_chunk(file: UploadFile = File(...), chunk_number: int = Form(...), total_chunks: int = Form(...)):
+    chunk_data = await file.read()
+    save_chunk(chunk_data, file.filename, chunk_number)
+    
+    if chunk_number == total_chunks - 1:
+        reassemble_file(file.filename, total_chunks)
+    
+    return JSONResponse(content={"message": "Chunk uploaded successfully"})
+
+# Wrap Streamlit app
+streamlit_app = WSGIMiddleware(app)
+
+def save_chunk(chunk, filename, chunk_number):
+    save_path = Path("uploaded_chunks")
+    save_path.mkdir(exist_ok=True)
+    file_path = save_path / f"{filename}.part{chunk_number}"
+    with file_path.open("wb") as f:
+        f.write(chunk)
+
+def reassemble_file(filename, total_chunks):
+    save_path = Path("uploaded_videos")
+    save_path.mkdir(exist_ok=True)
+    file_path = save_path / filename
+    
+    with file_path.open("wb") as outfile:
+        for i in range(total_chunks):
+            chunk_path = Path("uploaded_chunks") / f"{filename}.part{i}"
+            with chunk_path.open("rb") as infile:
+                outfile.write(infile.read())
+            os.remove(chunk_path)
 
 def get_uploaded_videos():
     video_dir = Path("uploaded_videos")
@@ -169,37 +163,95 @@ def generate_thumbnail(video_path):
 def main():
     st.markdown("<h1 style='text-align: center;'>UltraVideoSpace</h1>", unsafe_allow_html=True)
 
-    uploaded_file = st.file_uploader("", type=['mp4', 'mkv', 'avi', 'mov', 'mpeg4'], key="video_uploader")
-    
-    upload_placeholder = st.empty()
-    with upload_placeholder.container():
-        st.markdown("""
-            <div class="upload-zone">
-                <h3>Drag and drop your video here</h3>
-                <p>or click to select files</p>
-                <p>Supported formats: MP4, MKV, AVI, MOV, MPEG4</p>
-                <p>Maximum file size: 10GB</p>
-            </div>
-        """, unsafe_allow_html=True)
-
-    if uploaded_file is not None:
-        upload_placeholder.empty()
-        st.write(f"Uploading: {uploaded_file.name}")
-        file_size_mb = uploaded_file.size / (1024 * 1024)
-        file_details = {
-            "FileName": uploaded_file.name,
-            "FileType": uploaded_file.type,
-            "FileSize": f"{file_size_mb:.2f} MB"
+    st.markdown("""
+    <div class="upload-zone" id="drop-zone">
+        <h3>Drag and drop your video here</h3>
+        <p>or click to select files</p>
+        <p>Supported formats: MP4, MKV, AVI, MOV, MPEG4</p>
+        <p>No file size limit (files will be split if larger than 200MB)</p>
+    </div>
+    <div id="file-upload-status"></div>
+    """, unsafe_allow_html=True)
+
+    # File uploader and chunked upload handling
+    st.markdown("""
+    <script>
+    const dropZone = document.getElementById('drop-zone');
+    const statusDiv = document.getElementById('file-upload-status');
+
+    dropZone.addEventListener('dragover', (e) => {
+        e.preventDefault();
+        dropZone.style.backgroundColor = 'rgba(229, 9, 20, 0.1)';
+    });
+
+    dropZone.addEventListener('dragleave', (e) => {
+        e.preventDefault();
+        dropZone.style.backgroundColor = '';
+    });
+
+    dropZone.addEventListener('drop', (e) => {
+        e.preventDefault();
+        dropZone.style.backgroundColor = '';
+        const file = e.dataTransfer.files[0];
+        uploadFile(file);
+    });
+
+    dropZone.addEventListener('click', () => {
+        const input = document.createElement('input');
+        input.type = 'file';
+        input.accept = '.mp4,.mkv,.avi,.mov,.mpeg4';
+        input.onchange = (e) => {
+            const file = e.target.files[0];
+            uploadFile(file);
+        };
+        input.click();
+    });
+
+    function uploadFile(file) {
+        const chunkSize = 200 * 1024 * 1024; // 200MB
+        const totalChunks = Math.ceil(file.size / chunkSize);
+        let currentChunk = 0;
+
+        statusDiv.innerHTML = `Uploading ${file.name} (0/${totalChunks} chunks)`;
+
+        function uploadNextChunk() {
+            const start = currentChunk * chunkSize;
+            const end = Math.min(start + chunkSize, file.size);
+            const chunk = file.slice(start, end);
+
+            const formData = new FormData();
+            formData.append('file', chunk, file.name);
+            formData.append('chunk_number', currentChunk);
+            formData.append('total_chunks', totalChunks);
+
+            fetch('/upload_chunk', {
+                method: 'POST',
+                body: formData
+            })
+            .then(response => response.json())
+            .then(data => {
+                currentChunk++;
+                statusDiv.innerHTML = `Uploading ${file.name} (${currentChunk}/${totalChunks} chunks)`;
+                if (currentChunk < totalChunks) {
+                    uploadNextChunk();
+                } else {
+                    statusDiv.innerHTML = `${file.name} uploaded successfully!`;
+                    setTimeout(() => {
+                        window.location.reload();
+                    }, 2000);
+                }
+            })
+            .catch(error => {
+                statusDiv.innerHTML = `Error uploading ${file.name}: ${error}`;
+            });
         }
-        st.write(file_details)
 
-        if file_size_mb > 10 * 1024:  # If file size is greater than 10GB
-            st.error("File size exceeds the 10GB limit. Please upload a smaller file.")
-        elif save_uploaded_file(uploaded_file):
-            st.success(f"File {uploaded_file.name} successfully uploaded!")
-        else:
-            st.error("Failed to upload file.")
+        uploadNextChunk();
+    }
+    </script>
+    """, unsafe_allow_html=True)
 
+    # Display uploaded videos
     st.markdown("<h2 style='text-align: center; margin-top: 40px;'>Your Video Collection</h2>", unsafe_allow_html=True)
     videos = get_uploaded_videos()