Update app.py

app.py

@@ -108,8 +108,15 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
+# Constants
+MAX_FILE_SIZE = 10 * 1024 * 1024 * 1024  # 10GB in bytes
+
 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
@@ -145,15 +152,18 @@ def get_uploaded_videos():
     return [f for f in video_dir.glob("*") if f.suffix.lower() in ['.mp4', '.mkv', '.avi', '.mov', '.mpeg4']]
 
 def generate_thumbnail(video_path):
-    video = cv2.VideoCapture(str(video_path))
-    success, image = video.read()
-    if success:
-        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-        image = Image.fromarray(image)
-        image.thumbnail((300, 168))
-        buffered = io.BytesIO()
-        image.save(buffered, format="PNG")
-        return base64.b64encode(buffered.getvalue()).decode()
+    try:
+        video = cv2.VideoCapture(str(video_path))
+        success, image = video.read()
+        if success:
+            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+            image = Image.fromarray(image)
+            image.thumbnail((300, 168))
+            buffered = io.BytesIO()
+            image.save(buffered, format="PNG")
+            return base64.b64encode(buffered.getvalue()).decode()
+    except Exception as e:
+        st.warning(f"Could not generate thumbnail for {video_path.name}: {str(e)}")
     return None
 
 def main():
@@ -168,16 +178,24 @@ def main():
                 <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</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_details = {"FileName": uploaded_file.name, "FileType": uploaded_file.type, "FileSize": f"{uploaded_file.size / (1024*1024):.2f} MB"}
+        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.write(file_details)
-        if save_uploaded_file(uploaded_file):
+
+        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.")