Update app.py

app.py

@@ -15,7 +15,6 @@ import glob
 import os
 import zipfile
 
-# File type definitions
 FILE_TYPES = {
     "md": "📝 Markdown",
     "txt": "📄 Text",
@@ -23,20 +22,26 @@ FILE_TYPES = {
     "csv": "📊 CSV",
     "xlsx": "📗 Excel",
     "yaml": "⚙️ YAML",
-    "xml": "🔗 XML"
+    "xml": "🔗 XML",
+    "mp4": "🎬 MP4",
+    "webm": "🎬 WEBM",
+    "ogg": "🎬 OGG"
 }
 
-# Initialize session state
-for key in ['file_data', 'file_types', 'md_outline', 'rendered_content', 
-           'file_history', 'md_versions', 'md_files_history', 'combined_markdown',
-           'current_file', 'file_content']:
+for key in [
+    'file_data', 'file_types', 'md_outline', 'rendered_content', 'file_history',
+    'md_versions', 'md_files_history', 'combined_markdown', 'current_file',
+    'file_content'
+]:
     if key not in st.session_state:
-        st.session_state[key] = {} if key in ['file_data', 'file_types', 'md_outline', 
-                                            'rendered_content', 'md_versions'] else [] if key in ['file_history', 
-                                            'md_files_history'] else None if key in ['current_file', 'file_content'] else ""
+        if key in ['file_data', 'file_types', 'md_outline', 'rendered_content', 'md_versions']:
+            st.session_state[key] = {}
+        elif key in ['file_history', 'md_files_history']:
+            st.session_state[key] = []
+        else:
+            st.session_state[key] = ""
 
 def create_zip_of_files(files):
-    """Create a zip file containing all files"""
     zip_buffer = BytesIO()
     with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
         for file in files:
@@ -45,12 +50,11 @@ def create_zip_of_files(files):
     return zip_buffer.getvalue()
 
 def get_download_link(file_path):
-    """Create a download link for a file"""
-    if isinstance(file_path, bytes):  # For zip files
+    if isinstance(file_path, bytes):
         b64 = base64.b64encode(file_path).decode()
         href = f'data:application/zip;base64,{b64}'
         filename = 'all_files.zip'
-    else:  # For individual files
+    else:
         with open(file_path, 'r', encoding='utf-8') as f:
             content = f.read()
         b64 = base64.b64encode(content.encode()).decode()
@@ -59,26 +63,26 @@ def get_download_link(file_path):
     return f'<a href="{href}" download="{filename}">{filename}</a>'
 
 def load_file(filepath):
-    """Load file content"""
     with open(filepath, 'r', encoding='utf-8') as f:
         return f.read()
 
 def encode_content(content):
-    """Encode content to base64"""
     return base64.b64encode(content.encode()).decode()
 
 def decode_content(encoded_content):
-    """Decode content from base64"""
     return base64.b64decode(encoded_content.encode()).decode()
 
 def parse_markdown_outline(content):
-    """Generate an outline from markdown content"""
-    return [{'level': len(line.split()[0]), 'title': line.strip('#').strip(), 
-             'indent': '    ' * (len(line.split()[0]) - 1)} 
-            for line in content.split('\n') if line.strip().startswith('#')]
+    return [
+        {
+            'level': len(line.split()[0]),
+            'title': line.strip('#').strip(),
+            'indent': '    ' * (len(line.split()[0]) - 1)
+        }
+        for line in content.split('\n') if line.strip().startswith('#')
+    ]
 
 def combine_markdown_files():
-    """Combine markdown files into a single document"""
     all_files = glob.glob("*.md")
     combined = []
     for filepath in sorted(all_files):
@@ -87,8 +91,23 @@ def combine_markdown_files():
             combined.append(f"# {filepath}\n\n{content}\n\n---\n\n")
     return "".join(combined)
 
+def combine_video_files():
+    video_files = [f for f in glob.glob("*") if f.split(".")[-1].lower() in ["mp4", "webm", "ogg"]]
+    video_html = []
+    for vf in video_files:
+        with open(vf, "rb") as f:
+            b64_video = base64.b64encode(f.read()).decode("utf-8")
+        ext = vf.split(".")[-1].lower()
+        mime_type = "video/mp4" if ext == "mp4" else f"video/{ext}"
+        video_html.append(f"""
+        <video width="640" height="480" controls autoplay muted loop>
+            <source src="data:{mime_type};base64,{b64_video}" type="{mime_type}">
+        </video>
+        <br><hr>
+        """)
+    return "".join(video_html)
+
 def display_file_manager():
-    """Display file management sidebar with guaranteed unique button keys."""
     st.sidebar.title("📁 File Management")
     all_files = glob.glob("*.md")
     all_files.sort(reverse=True)
@@ -97,7 +116,7 @@ def display_file_manager():
     with col1:
         if st.button("🗑 Delete All", key="delete_all_files_button"):
             for file in all_files:
-                if file != "README.md":  # Protect README.md
+                if file != "README.md":
                     os.remove(file)
             st.rerun()
     with col2:
@@ -107,7 +126,6 @@ def display_file_manager():
     
     st.sidebar.markdown("---")
     
-    # Create unique keys using file attributes
     for idx, file in enumerate(all_files):
         file_stat = os.stat(file)
         unique_id = f"{idx}_{file_stat.st_size}_{file_stat.st_mtime}"
@@ -129,12 +147,10 @@ def display_file_manager():
                 st.rerun()
 
 def create_markdown_tabs(content, filename):
-    """Create tabs for markdown content viewing and editing"""
     tab1, tab2 = st.tabs(["📝 Editor", "👀 Preview"])
     
     with tab1:
-        edited_content = st.text_area("Edit your markdown", content, height=300, 
-                                    key=f"edit_{filename}")
+        edited_content = st.text_area("Edit your markdown", content, height=300, key=f"edit_{filename}")
         if edited_content != content:
             st.session_state.file_data[filename] = edited_content
             content = edited_content
@@ -150,7 +166,6 @@ def create_markdown_tabs(content, filename):
     return content
 
 def read_file_content(uploaded_file):
-    """Smart file reader with enhanced markdown handling"""
     file_type = uploaded_file.name.split('.')[-1].lower()
     try:
         if file_type == 'md':
@@ -159,27 +174,32 @@ def read_file_content(uploaded_file):
                 f.write(content)
             return content, "md"
         elif file_type in ['csv', 'xlsx']:
-            df = pd.read_csv(uploaded_file) if file_type == 'csv' else pd.read_excel(uploaded_file)
+            if file_type == 'csv':
+                df = pd.read_csv(uploaded_file)
+            else:
+                df = pd.read_excel(uploaded_file)
             return df.to_string(), file_type
         elif file_type == 'json':
             return json.dumps(json.load(uploaded_file), indent=2), "json"
         elif file_type == 'yaml':
             return yaml.dump(yaml.safe_load(uploaded_file)), "yaml"
+        elif file_type in ['mp4', 'webm', 'ogg']:
+            with open(uploaded_file.name, 'wb') as f:
+                f.write(uploaded_file.getvalue())
+            return None, file_type
         else:
-            return uploaded_file.getvalue().decode(), "txt"
+            content = uploaded_file.getvalue().decode()
+            return content, "txt"
     except Exception as e:
         st.error(f"🚨 Error reading {uploaded_file.name}: {str(e)}")
         return None, None
 
 def main():
     st.set_page_config(page_title="Markdown Handler", layout="wide")
-    
-    # Display file manager in sidebar
     display_file_manager()
     
     st.title("📚✨ Super Smart File Handler with Markdown Magic! ✨📚")
     
-    # Add tabs for different views
     upload_tab, book_tab = st.tabs(["📤 File Upload", "📖 Book View"])
     
     with upload_tab:
@@ -187,7 +207,7 @@ def main():
         
         with col1:
             single_uploaded_file = st.file_uploader(
-                "📤 Upload single file", 
+                "📤 Upload single file",
                 type=list(FILE_TYPES.keys()),
                 help="Supports: " + ", ".join([f"{v} (.{k})" for k, v in FILE_TYPES.items()]),
                 key="single_uploader"
@@ -198,24 +218,24 @@ def main():
                 "📚 Upload multiple files",
                 type=list(FILE_TYPES.keys()),
                 accept_multiple_files=True,
-                help="Upload multiple files to view as a book",
+                help="Upload multiple files to view as a book or video playlist",
                 key="multiple_uploader"
             )
         
-        # Process uploads
         if single_uploaded_file:
             content, file_type = read_file_content(single_uploaded_file)
-            if content is not None:
+            if content is not None and file_type != "txt":
                 st.success(f"🎉 Loaded {FILE_TYPES.get(file_type, '📄')} file: {single_uploaded_file.name}")
+            elif content is not None:
+                st.success(f"🎉 Loaded file: {single_uploaded_file.name}")
         
         if multiple_uploaded_files:
             for uploaded_file in multiple_uploaded_files:
                 content, file_type = read_file_content(uploaded_file)
-                if content is not None:
+                if content is not None or file_type in ['mp4','webm','ogg']:
                     st.success(f"🎉 Loaded {uploaded_file.name}")
             st.rerun()
         
-        # Show current file if selected
         if st.session_state.current_file and st.session_state.file_content:
             st.markdown("---")
             create_markdown_tabs(st.session_state.file_content, st.session_state.current_file)
@@ -227,6 +247,13 @@ def main():
             st.markdown(combined_content, unsafe_allow_html=True)
         else:
             st.info("Upload some markdown files to see them combined here!")
+        
+        st.markdown("## 🎞 Video Gallery")
+        video_playlist = combine_video_files()
+        if video_playlist:
+            st.markdown(video_playlist, unsafe_allow_html=True)
+        else:
+            st.info("No videos found. Upload MP4, WEBM, or OGG files to see them here!")
 
 if __name__ == "__main__":
-    main()
+    main()