Update app.py

app.py

@@ -26,6 +26,10 @@ import pandas as pd
 import plotly.express as px
 import markdown
 import zipfile
+import contextlib
+import threading
+import traceback
+from io import StringIO, BytesIO
 
 # Set up enhanced logging
 logging.basicConfig(
@@ -63,7 +67,6 @@ def ensure_packages():
         'matplotlib': '3.5.0',  # For Python script runner
         'seaborn': '0.11.2',  # For enhanced visualizations
         'scipy': '1.7.3',    # For scientific computations
-        'accelerate': '0.20.0',
     }
     
     with st.spinner("Checking required packages..."):
@@ -302,7 +305,7 @@ QUALITY_PRESETS = {
     "720p": {"resolution": "720p", "fps": "30"},
     "1080p": {"resolution": "1080p", "fps": "60"},
     "4K": {"resolution": "2160p", "fps": "60"},
-    "8K": {"resolution": "4320p", "fps": "60"}
+    "8K": {"resolution": "4320p", "fps": "60"}  # Added 8K option
 }
 
 # Animation speeds
@@ -510,7 +513,7 @@ def generate_manim_video(python_code, format_type, quality_preset, animation_spe
             "720p": "-qm",  # Medium quality
             "1080p": "-qh",  # High quality
             "4K": "-qk",     # 4K quality
-            "8K": "-qu"      # Ultra quality (8K)
+            "8K": "-qp"      # 8K quality (production quality)
         }
         quality_flag = quality_map.get(quality_preset, "-qm")
         
@@ -1170,7 +1173,7 @@ def create_timeline_editor(code):
             animation_steps = df.to_dict('records')
             new_code = generate_code_from_timeline(animation_steps, code)
             
-            st.success("Timeline updated! Click 'Refresh App' to see changes.")
+            st.success("Timeline updated! Code has been regenerated.")
             return new_code
     
     # Visual keyframe editor
@@ -1467,32 +1470,6 @@ def export_to_educational_format(video_data, format_type, animation_title, expla
         logger.error(traceback.format_exc())
         return None, None
 
-def process_multiple_images(uploaded_images):
-    """Process multiple uploaded images and return information about them"""
-    processed_images = []
-    
-    # Create a unique image directory if it doesn't exist
-    image_dir = os.path.join(os.getcwd(), "manim_assets", "images")
-    os.makedirs(image_dir, exist_ok=True)
-    
-    for img in uploaded_images:
-        # Generate a unique filename and save the image
-        file_extension = img.name.split(".")[-1]
-        unique_filename = f"image_{int(time.time())}_{len(processed_images)}.{file_extension}"
-        image_path = os.path.join(image_dir, unique_filename)
-        
-        with open(image_path, "wb") as f:
-            f.write(img.getvalue())
-        
-        # Store image info
-        processed_images.append({
-            "name": img.name, 
-            "path": image_path,
-            "preview": img
-        })
-    
-    return processed_images
-
 def main():
     # Initialize session state variables if they don't exist
     if 'init' not in st.session_state:
@@ -1511,13 +1488,12 @@ def main():
         st.session_state.custom_library_result = ""
         st.session_state.python_script = "import matplotlib.pyplot as plt\nimport numpy as np\n\n# Example: Create a simple plot\nx = np.linspace(0, 10, 100)\ny = np.sin(x)\n\nplt.figure(figsize=(10, 6))\nplt.plot(x, y, 'b-', label='sin(x)')\nplt.title('Sine Wave')\nplt.xlabel('x')\nplt.ylabel('sin(x)')\nplt.grid(True)\nplt.legend()\n"
         st.session_state.python_result = None
+        st.session_state.active_tab = 0  # Track currently active tab
         st.session_state.settings = {
             "quality": "720p",
             "format_type": "mp4",
             "animation_speed": "Normal"
         }
-        st.session_state.pending_changes = False  # Track if there are changes that need refresh
-        st.session_state.pending_temp_code = ""   # Store code changes before refresh
 
     # Page configuration with improved layout
     st.set_page_config(
@@ -1625,41 +1601,6 @@ def main():
         margin-top: 1rem;
         border-left: 4px solid #ef4444;
     }
-    
-    .refresh-button {
-        background-color: #4CAF50;
-        color: white;
-        padding: 10px 20px;
-        border: none;
-        border-radius: 5px;
-        cursor: pointer;
-        font-weight: bold;
-        margin: 10px 0;
-    }
-    
-    .refresh-button:hover {
-        background-color: #45a049;
-    }
-    
-    .image-grid {
-        display: grid;
-        grid-template-columns: repeat(auto-fill, minmax(150px, 1fr));
-        gap: 10px;
-        margin-top: 10px;
-    }
-    
-    .image-item {
-        border: 1px solid #ddd;
-        border-radius: 5px;
-        padding: 5px;
-        text-align: center;
-    }
-    
-    .image-item img {
-        max-width: 100%;
-        height: auto;
-        border-radius: 3px;
-    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -1684,18 +1625,11 @@ def main():
         st.session_state.ai_models = init_ai_models()
     
     # Create main tabs
-    tabs = st.tabs(["✨ Editor", "🤖 AI Assistant", "📚 LaTeX Formulas", "🎨 Assets", "🎞️ Timeline", "🎓 Educational Export", "🐍 Python Runner"])
+    tab_names = ["✨ Editor", "🤖 AI Assistant", "📚 LaTeX Formulas", "🎨 Assets", "🎞️ Timeline", "🎓 Educational Export", "🐍 Python Runner"]
+    tabs = st.tabs(tab_names)
     
-    # Sidebar for rendering settings, custom libraries, and refresh button
+    # Sidebar for rendering settings and custom libraries
     with st.sidebar:
-        # Add refresh button at the top of the sidebar
-        if st.button("🔄 Refresh App", key="refresh_button", help="Refresh the app to apply all changes", use_container_width=True):
-            st.success("App refreshed!")
-            if st.session_state.pending_changes:
-                st.session_state.code = st.session_state.pending_temp_code
-                st.session_state.temp_code = st.session_state.pending_temp_code
-                st.session_state.pending_changes = False
-            
         # Rendering settings section
         st.markdown("## ⚙️ Rendering Settings")
         
@@ -1775,36 +1709,31 @@ def main():
                 if uploaded_file:
                     code_content = uploaded_file.getvalue().decode("utf-8")
                     if code_content.strip():  # Only update if file has content
-                        st.session_state.pending_temp_code = code_content
-                        st.session_state.pending_changes = True
-                        st.info("Code loaded from file. Click 'Refresh App' to apply changes.")
+                        st.session_state.code = code_content
+                        st.session_state.temp_code = code_content
             
             # Code editor
             if ACE_EDITOR_AVAILABLE:
                 current_code = st.session_state.code if hasattr(st.session_state, 'code') and st.session_state.code else ""
-                temp_code = st_ace(
+                st.session_state.temp_code = st_ace(
                     value=current_code,
                     language="python",
                     theme="monokai",
                     min_lines=20,
                     key="ace_editor"
                 )
-                if temp_code != current_code:
-                    st.session_state.pending_temp_code = temp_code
-                    st.session_state.pending_changes = True
-                    st.info("Code changes detected. Click 'Refresh App' to apply changes.")
             else:
                 current_code = st.session_state.code if hasattr(st.session_state, 'code') and st.session_state.code else ""
-                temp_code = st.text_area(
+                st.session_state.temp_code = st.text_area(
                     "Manim Python Code",
                     value=current_code,
                     height=400,
                     key="code_textarea"
                 )
-                if temp_code != current_code:
-                    st.session_state.pending_temp_code = temp_code
-                    st.session_state.pending_changes = True
-                    st.info("Code changes detected. Click 'Refresh App' to apply changes.")
+            
+            # Update code in session state if it changed
+            if st.session_state.temp_code != st.session_state.code:
+                st.session_state.code = st.session_state.temp_code
             
             # Generate button (use a form to prevent page reloads)
             generate_btn = st.button("🚀 Generate Animation", use_container_width=True, key="generate_btn")
@@ -1825,9 +1754,8 @@ class MyScene(Scene):
         self.wait(2)
 """
                         st.session_state.code += default_scene
-                        st.session_state.pending_temp_code = st.session_state.code
-                        st.session_state.pending_changes = True
-                        st.warning("No scene class found. Added a default scene. Click 'Refresh App' to see changes.")
+                        st.session_state.temp_code = st.session_state.code
+                        st.warning("No scene class found. Added a default scene.")
                         
                     with st.spinner("Generating animation..."):
                         video_data, status = generate_manim_video(
@@ -1915,7 +1843,7 @@ class MyScene(Scene):
                         1. **Syntax Errors**: Check your Python code for any syntax issues
                         2. **Missing Scene Class**: Ensure your code contains a scene class that extends Scene
                         3. **High Resolution Issues**: Try a lower quality preset for complex animations
-                        4. **Memory Issues**: For 4K or 8K animations, reduce complexity or try again
+                        4. **Memory Issues**: For 4K animations, reduce complexity or try again
                         5. **Format Issues**: Some formats require specific Manim configurations
                         6. **GIF Generation**: If GIF doesn't work, try MP4 and we'll convert it automatically
                         
@@ -1987,9 +1915,11 @@ class MyScene(Scene):
                 col_ai1, col_ai2 = st.columns(2)
                 with col_ai1:
                     if st.button("Use This Code", key="use_gen_code"):
-                        st.session_state.pending_temp_code = st.session_state.generated_code
-                        st.session_state.pending_changes = True
-                        st.info("AI code ready to use. Click 'Refresh App' to apply it to the editor.")
+                        st.session_state.code = st.session_state.generated_code
+                        st.session_state.temp_code = st.session_state.generated_code
+                        # Switch to Editor tab
+                        st.session_state.active_tab = 0
+                        st.rerun()
                 
                 with col_ai2:
                     if st.button("Render Preview", key="render_preview"):
@@ -2032,9 +1962,8 @@ class MyScene(Scene):
                 key="latex_input"
             )
             
-            # Update session state without auto-refresh
-            if latex_input != st.session_state.latex_formula:
-                st.session_state.latex_formula = latex_input
+            # Update session state
+            st.session_state.latex_formula = latex_input
             
             # Common LaTeX formulas library
             st.markdown("#### Formula Library")
@@ -2082,9 +2011,10 @@ class MyScene(Scene):
                 with formula_tabs[i]:
                     for formula in formulas:
                         if st.button(formula["name"], key=f"latex_{formula['name']}"):
+                            # Insert formula into the text area
                             st.session_state.latex_formula = formula["latex"]
-                            # No automatic rerun - user needs to click refresh
-                            st.info(f"Added '{formula['name']}'. Click 'Refresh App' to update the preview.")
+                            # Rerun to update the text area and preview
+                            st.rerun()
             
             # LaTeX code snippet
             st.markdown("#### Manim Code Snippet")
@@ -2122,10 +2052,13 @@ self.wait(2)
                                     # If we didn't find content, append to the end with default indentation
                                     lines.append("        " + "\n        ".join(manim_latex_code.strip().split("\n")))
                                 
-                                new_code = "\n".join(lines)
-                                st.session_state.pending_temp_code = new_code
-                                st.session_state.pending_changes = True
-                                st.success("LaTeX formula prepared for insertion. Click 'Refresh App' to update the editor.")
+                                st.session_state.code = "\n".join(lines)
+                                st.session_state.temp_code = st.session_state.code
+                                st.success("LaTeX formula inserted into the editor!")
+                                
+                                # Switch to Editor tab
+                                st.session_state.active_tab = 0
+                                st.rerun()
                         else:
                             st.warning("Could not find 'construct' method in your code. Please add a scene class first.")
                     else:
@@ -2139,9 +2072,13 @@ class LatexScene(Scene):
         self.play(Write(formula))
         self.wait(2)
 """
-                        st.session_state.pending_temp_code = basic_scene
-                        st.session_state.pending_changes = True
-                        st.success("Created new scene with your LaTeX formula. Click 'Refresh App' to update the editor.")
+                        st.session_state.code = basic_scene
+                        st.session_state.temp_code = basic_scene
+                        st.success("Created new scene with your LaTeX formula!")
+                        
+                        # Switch to Editor tab
+                        st.session_state.active_tab = 0
+                        st.rerun()
         
         with col_latex2:
             # LaTeX preview
@@ -2183,79 +2120,102 @@ class LatexScene(Scene):
             st.markdown("#### 📸 Image Assets")
             st.markdown("Upload images to use in your animations:")
             
-            # Multiple image uploader
-            uploaded_images = st.file_uploader("Upload Images", type=["jpg", "png", "jpeg", "svg"], accept_multiple_files=True, key="image_uploader_tab")
+            # Allow multiple image uploads
+            uploaded_images = st.file_uploader(
+                "Upload Images", 
+                type=["jpg", "png", "jpeg", "svg"], 
+                accept_multiple_files=True,
+                key="image_uploader_tab"
+            )
             
             if uploaded_images:
-                # Process all uploaded images
-                with st.spinner("Processing uploaded images..."):
-                    processed_images = process_multiple_images(uploaded_images)
+                # Create a unique image directory if it doesn't exist
+                image_dir = os.path.join(os.getcwd(), "manim_assets", "images")
+                os.makedirs(image_dir, exist_ok=True)
+                
+                # Process each uploaded image
+                for uploaded_image in uploaded_images:
+                    # Generate a unique filename and save the image
+                    file_extension = uploaded_image.name.split(".")[-1]
+                    unique_filename = f"image_{int(time.time())}_{uuid.uuid4().hex[:8]}.{file_extension}"
+                    image_path = os.path.join(image_dir, unique_filename)
                     
-                    # Add to session state
+                    with open(image_path, "wb") as f:
+                        f.write(uploaded_image.getvalue())
+                    
+                    # Store the path in session state
                     if "image_paths" not in st.session_state:
                         st.session_state.image_paths = []
                     
-                    st.session_state.image_paths.extend(processed_images)
-                    
-                    # Show preview of uploaded images
-                    st.success(f"Successfully uploaded {len(processed_images)} images")
+                    # Check if this image was already added
+                    # Check if this image was already added
+                    image_already_added = False
+                    for img in st.session_state.image_paths:
+                        if img["name"] == uploaded_image.name:
+                            image_already_added = True
+                            break
                     
-                    # Display image grid
-                    st.markdown("#### New Images")
-                    image_cols = st.columns(min(4, len(processed_images)))
-                    for i, img_info in enumerate(processed_images):
-                        with image_cols[i % len(image_cols)]:
-                            st.image(img_info["preview"], caption=img_info["name"], width=150)
+                    if not image_already_added:
+                        st.session_state.image_paths.append({
+                            "name": uploaded_image.name, 
+                            "path": image_path
+                        })
+                
+                # Display uploaded images in a grid
+                st.markdown("##### Uploaded Images:")
+                image_cols = st.columns(3)
+                
+                for i, img_info in enumerate(st.session_state.image_paths[-len(uploaded_images):]):
+                    with image_cols[i % 3]:
+                        try:
+                            img = Image.open(img_info["path"])
+                            st.image(img, caption=img_info["name"], width=150)
                             
-                    # Show code to use images
-                    if processed_images:
-                        st.markdown("#### Code to use these images:")
-                        image_code_samples = []
-                        for i, img_info in enumerate(processed_images):
-                            image_code = f"""
-# Load and display image {i+1}: {img_info['name']}
-image{i+1} = ImageMobject(r"{img_info['path']}")
-image{i+1}.scale(2)  # Adjust size as needed
-self.play(FadeIn(image{i+1}))
+                            # Show code snippet for this specific image
+                            if st.button(f"Use {img_info['name']}", key=f"use_img_{i}"):
+                                image_code = f"""
+# Load and display image
+image = ImageMobject(r"{img_info['path']}")
+image.scale(2)  # Adjust size as needed
+self.play(FadeIn(image))
 self.wait(1)
 """
-                            image_code_samples.append(image_code)
-                        
-                        all_images_code = "\n".join(image_code_samples)
-                        st.code(all_images_code, language="python")
-                        
-                        if st.button("Insert All Images Code", key="insert_images_code_btn"):
-                            if not st.session_state.code:
-                                base_code = """from manim import *
+                                if not st.session_state.code:
+                                    base_code = """from manim import *
 
 class ImageScene(Scene):
     def construct(self):
 """
-                                indented_code = "\n        ".join(all_images_code.split("\n"))
-                                new_code = base_code + "\n        " + indented_code
-                            else:
-                                new_code = st.session_state.code + "\n" + all_images_code
-                            
-                            st.session_state.pending_temp_code = new_code
-                            st.session_state.pending_changes = True
-                            st.success("Image code prepared for insertion. Click 'Refresh App' to update the editor.")
+                                    st.session_state.code = base_code + "\n        " + image_code.replace("\n", "\n        ")
+                                else:
+                                    st.session_state.code += "\n" + image_code
+                                
+                                st.session_state.temp_code = st.session_state.code
+                                st.success(f"Added {img_info['name']} to your code!")
+                                
+                                # Switch to Editor tab
+                                st.session_state.active_tab = 0
+                                st.rerun()
+                        except Exception as e:
+                            st.error(f"Error loading image {img_info['name']}: {e}")
             
             # Display previously uploaded images
             if st.session_state.image_paths:
-                st.markdown("#### All Available Images")
-                
-                # Display in a grid format
-                st.markdown('<div class="image-grid">', unsafe_allow_html=True)
-                
-                for idx, img_info in enumerate(st.session_state.image_paths):
-                    st.markdown(f"""
-                    <div class="asset-card">
-                        <p><strong>{img_info['name']}</strong></p>
-                        <p class="small-text">Path: {img_info['path']}</p>
-                    </div>
-                    """, unsafe_allow_html=True)
-                
-                st.markdown('</div>', unsafe_allow_html=True)
+                with st.expander("Previously Uploaded Images"):
+                    # Group images by 3 in each row
+                    for i in range(0, len(st.session_state.image_paths), 3):
+                        prev_cols = st.columns(3)
+                        for j in range(3):
+                            if i+j < len(st.session_state.image_paths):
+                                img_info = st.session_state.image_paths[i+j]
+                                with prev_cols[j]:
+                                    try:
+                                        img = Image.open(img_info["path"])
+                                        st.image(img, caption=img_info["name"], width=100)
+                                        st.markdown(f"<div class='small-text'>Path: {img_info['path']}</div>", unsafe_allow_html=True)
+                                    except:
+                                        st.markdown(f"**{img_info['name']}**")
+                                        st.markdown(f"<div class='small-text'>Path: {img_info['path']}</div>", unsafe_allow_html=True)
             
         with asset_col2:
             # Audio uploader section
@@ -2341,8 +2301,8 @@ class YourScene(Scene):
         
         # If code was modified by the timeline editor, update the session state
         if updated_code != st.session_state.code:
-            st.session_state.pending_temp_code = updated_code
-            st.session_state.pending_changes = True
+            st.session_state.code = updated_code
+            st.session_state.temp_code = updated_code
 
     # EDUCATIONAL EXPORT TAB
     with tabs[5]:
@@ -2626,11 +2586,13 @@ plt.legend()
         
         # Python code editor
         if selected_example != "Select an example..." and selected_example in example_scripts:
-            st.session_state.python_script = example_scripts[selected_example]
+            python_code = example_scripts[selected_example]
+        else:
+            python_code = st.session_state.python_script
         
         if ACE_EDITOR_AVAILABLE:
             python_code = st_ace(
-                value=st.session_state.python_script,
+                value=python_code,
                 language="python",
                 theme="monokai",
                 min_lines=15,
@@ -2639,12 +2601,12 @@ plt.legend()
         else:
             python_code = st.text_area(
                 "Python Code",
-                value=st.session_state.python_script,
+                value=python_code,
                 height=400,
                 key="python_textarea"
             )
         
-        # Update session state
+        # Store script in session state (without clearing existing code)
         st.session_state.python_script = python_code
         
         # Check for input() calls
@@ -2707,20 +2669,25 @@ self.wait(1)
 """
                                 # Insert into editor code
                                 if st.session_state.code:
-                                    new_code = st.session_state.code + "\n" + plot_code
-                                    st.session_state.pending_temp_code = new_code
-                                    st.session_state.pending_changes = True
-                                    st.success(f"Plot {i+1} added to your animation code! Click 'Refresh App' to update the editor.")
+                                    st.session_state.code += "\n" + plot_code
+                                    st.session_state.temp_code = st.session_state.code
+                                    st.success(f"Plot {i+1} added to your animation code!")
+                                    # Switch to Editor tab
+                                    st.session_state.active_tab = 0
+                                    st.rerun()
                                 else:
                                     basic_scene = f"""from manim import *
 
 class PlotScene(Scene):
     def construct(self):
-{plot_code}
+        {plot_code}
 """
-                                    st.session_state.pending_temp_code = basic_scene
-                                    st.session_state.pending_changes = True
-                                    st.success(f"Created new scene with Plot {i+1}! Click 'Refresh App' to update the editor.")
+                                    st.session_state.code = basic_scene
+                                    st.session_state.temp_code = basic_scene
+                                    st.success(f"Created new scene with Plot {i+1}!")
+                                    # Switch to Editor tab
+                                    st.session_state.active_tab = 0
+                                    st.rerun()
             
             # Provide option to save the script
             if st.button("📄 Save This Script", key="save_script_btn"):
@@ -2795,6 +2762,13 @@ class PlotScene(Scene):
                 self.wait(2)
         ```
         """)
+    
+    # Restore active tab if specified
+    if st.session_state.active_tab is not None and st.session_state.active_tab < len(tabs):
+        for i, tab in enumerate(tabs):
+            if i == st.session_state.active_tab:
+                tab.active = True
+                break
 
 if __name__ == "__main__":
     main()