Update app.py

app.py

@@ -49,6 +49,39 @@ except ImportError:
     ACE_EDITOR_AVAILABLE = False
     logger.warning("streamlit-ace not available, falling back to standard text editor")
 
+# New functions for accessing secrets and password verification
+def get_secret(secret_name):
+    """Retrieve a secret from HuggingFace Spaces environment variables"""
+    secret_value = os.environ.get(secret_name)
+    if not secret_value:
+        logger.warning(f"Secret '{secret_name}' not found")
+        return None
+    return secret_value
+
+def check_password():
+    """Returns True if the user entered the correct password"""
+    # Get the password from secrets
+    correct_password = get_secret("password")
+    if not correct_password:
+        st.error("Admin password not configured in HuggingFace Spaces secrets")
+        return False
+        
+    # Password input
+    if "password_entered" not in st.session_state:
+        st.session_state.password_entered = False
+        
+    if not st.session_state.password_entered:
+        password = st.text_input("Enter password to access AI features", type="password")
+        if password:
+            if password == correct_password:
+                st.session_state.password_entered = True
+                return True
+            else:
+                st.error("Incorrect password")
+                return False
+        return False
+    return True
+
 def ensure_packages():
     required_packages = {
         'manim': '0.17.3',
@@ -67,6 +100,7 @@ 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
+        'huggingface_hub': '0.16.0',  # For Hugging Face API
     }
     
     with st.spinner("Checking required packages..."):
@@ -108,6 +142,8 @@ def ensure_packages():
                     import seaborn
                 elif package == 'scipy':
                     import scipy
+                elif package == 'huggingface_hub':
+                    import huggingface_hub
             except ImportError:
                 missing_packages[package] = version
         
@@ -202,19 +238,38 @@ def install_custom_packages(package_list):
     return success, "\n".join(results)
 
 @st.cache_resource(ttl=3600)
-def init_ai_models():
+def init_ai_models(model_name=None):
     try:        
-        with st.spinner("Loading AI models - this may take a moment..."):
+        # Get the GitHub token from secrets
+        github_token = get_secret("github_token_api")
+        if not github_token:
+            st.error("GitHub API token not configured in HuggingFace Spaces secrets")
+            return None
+        
+        # Use default model if none specified
+        if not model_name:
+            model_name = "deepseek-ai/deepseek-coder-1.3b-base"
+            
+        with st.spinner(f"Loading AI model: {model_name} - this may take a moment..."):
+            # Configure the Hugging Face API with the token
+            from huggingface_hub import login
+            login(token=github_token)
+            
+            # Now initialize the model with the token
             code_model = pipeline(
                 "text-generation",
-                model="deepseek-ai/deepseek-coder-1.3b-base",
+                model=model_name,
+                token=github_token,  # Pass the token for authentication
                 trust_remote_code=True
             )
+            
             if code_model is None:
                 st.error("Failed to initialize code model")
                 return None
+                
             return {
                 "code_model": code_model,
+                "model_name": model_name,
                 "last_loaded": datetime.now().isoformat()
             }
     except Exception as e:
@@ -1494,6 +1549,8 @@ def main():
             "format_type": "mp4",
             "animation_speed": "Normal"
         }
+        st.session_state.password_entered = False  # Track password authentication
+        st.session_state.custom_model = "deepseek-ai/deepseek-coder-1.3b-base"  # Default model
 
     # Page configuration with improved layout
     st.set_page_config(
@@ -1619,10 +1676,6 @@ def main():
         else:
             st.error("Failed to install required packages. Please try again.")
             st.stop()
-
-    # Initialize AI models if needed
-    if st.session_state.ai_models is None or not check_model_freshness():
-        st.session_state.ai_models = init_ai_models()
     
     # Create main tabs
     tab_names = ["✨ Editor", "🤖 AI Assistant", "📚 LaTeX Formulas", "🎨 Assets", "🎞️ Timeline", "🎓 Educational Export", "🐍 Python Runner"]
@@ -1865,85 +1918,144 @@ class MyScene(Scene):
     with tabs[1]:
         st.markdown("### 🤖 AI Animation Assistant")
         
-        if st.session_state.ai_models:
-            # AI code generation
-            st.markdown("<div class='card'>", unsafe_allow_html=True)
-            st.markdown("#### Generate Animation from Description")
-            st.write("Describe the animation you want to create, or provide partial code to complete.")
-            
-            # Predefined animation ideas dropdown
-            animation_ideas = [
-                "Select an idea...",
-                "Create a 3D animation showing a sphere morphing into a torus",
-                "Show a visual proof of the Pythagorean theorem",
-                "Visualize a Fourier transform converting a signal from time domain to frequency domain",
-                "Create an animation explaining neural network forward propagation",
-                "Illustrate the concept of integration with area under a curve"
+        # Check password before allowing access
+        if check_password():
+            # Model selection - only visible after authentication
+            st.markdown("#### Model Selection")
+            
+            # Predefined popular models
+            popular_models = [
+                "Select a predefined model...",
+                "deepseek-ai/deepseek-coder-1.3b-base",
+                "mistralai/Mistral-7B-Instruct-v0.2",
+                "Phind/Phind-CodeLlama-34B-v2",
+                "facebook/incoder-1B", 
+                "Salesforce/codegen2-1B",
             ]
             
-            selected_idea = st.selectbox(
-                "Try one of these ideas", 
-                options=animation_ideas
-            )
-            
-            prompt_value = selected_idea if selected_idea != "Select an idea..." else ""
-            
-            code_input = st.text_area(
-                "Your Prompt or Code",
-                value=prompt_value,
-                placeholder="Example: Create an animation that shows a circle morphing into a square while changing color from red to blue",
-                height=150
+            custom_model_option = st.radio(
+                "Choose model option:",
+                ["Use predefined model", "Enter custom model name"]
             )
             
-            if st.button("Generate Animation Code", key="gen_ai_code"):
-                if code_input:
-                    with st.spinner("AI is generating your animation code..."):
-                        response = suggest_code_completion(code_input, st.session_state.ai_models)
-                        if response:
-                            st.session_state.generated_code = response
-                else:
-                    st.warning("Please enter a description or prompt first")
+            if custom_model_option == "Use predefined model":
+                selected_model = st.selectbox(
+                    "Select a model:", 
+                    options=popular_models
+                )
+                
+                if selected_model != "Select a predefined model...":
+                    st.session_state.custom_model = selected_model
+            else:
+                custom_model_name = st.text_input(
+                    "Enter custom model name/path:", 
+                    value=st.session_state.custom_model,
+                    placeholder="e.g., organization/model-name"
+                )
+                
+                if custom_model_name:
+                    st.session_state.custom_model = custom_model_name
             
-            st.markdown("</div>", unsafe_allow_html=True)
+            # Display currently selected model
+            st.info(f"Currently selected model: {st.session_state.custom_model}")
             
-            # AI generated code display and actions
-            if st.session_state.generated_code:
+            # Only initialize models if password check passes and we have a model name
+            if st.session_state.custom_model:
+                model_changed = False
+                
+                # Check if we need to reinitialize the model (if model changed)
+                if (st.session_state.ai_models is not None and 
+                    'model_name' in st.session_state.ai_models and
+                    st.session_state.ai_models['model_name'] != st.session_state.custom_model):
+                    model_changed = True
+                
+                # Initialize model if needed
+                if (st.session_state.ai_models is None or 
+                    not check_model_freshness() or
+                    model_changed):
+                    st.session_state.ai_models = init_ai_models(st.session_state.custom_model)
+            
+            if st.session_state.ai_models:
+                # AI code generation
                 st.markdown("<div class='card'>", unsafe_allow_html=True)
-                st.markdown("#### Generated Animation Code")
-                st.code(st.session_state.generated_code, language="python")
+                st.markdown("#### Generate Animation from Description")
+                st.write("Describe the animation you want to create, or provide partial code to complete.")
                 
-                col_ai1, col_ai2 = st.columns(2)
-                with col_ai1:
-                    if st.button("Use This Code", key="use_gen_code"):
-                        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()
+                # Predefined animation ideas dropdown
+                animation_ideas = [
+                    "Select an idea...",
+                    "Create a 3D animation showing a sphere morphing into a torus",
+                    "Show a visual proof of the Pythagorean theorem",
+                    "Visualize a Fourier transform converting a signal from time domain to frequency domain",
+                    "Create an animation explaining neural network forward propagation",
+                    "Illustrate the concept of integration with area under a curve"
+                ]
+                
+                selected_idea = st.selectbox(
+                    "Try one of these ideas", 
+                    options=animation_ideas
+                )
+                
+                prompt_value = selected_idea if selected_idea != "Select an idea..." else ""
+                
+                code_input = st.text_area(
+                    "Your Prompt or Code",
+                    value=prompt_value,
+                    placeholder="Example: Create an animation that shows a circle morphing into a square while changing color from red to blue",
+                    height=150
+                )
+                
+                if st.button("Generate Animation Code", key="gen_ai_code"):
+                    if code_input:
+                        with st.spinner("AI is generating your animation code..."):
+                            response = suggest_code_completion(code_input, st.session_state.ai_models)
+                            if response:
+                                st.session_state.generated_code = response
+                    else:
+                        st.warning("Please enter a description or prompt first")
                 
-                with col_ai2:
-                    if st.button("Render Preview", key="render_preview"):
-                        with st.spinner("Rendering preview..."):
-                            video_data, status = generate_manim_video(
-                                st.session_state.generated_code, 
-                                "mp4", 
-                                "480p",  # Use lowest quality for preview
-                                ANIMATION_SPEEDS["Normal"]
-                            )
-                            
-                            if video_data:
-                                st.video(video_data)
-                                st.download_button(
-                                    label="Download Preview",
-                                    data=video_data,
-                                    file_name=f"manim_preview_{int(time.time())}.mp4",
-                                    mime="video/mp4"
-                                )
-                            else:
-                                st.error(f"Failed to generate preview: {status}")
                 st.markdown("</div>", unsafe_allow_html=True)
+                
+                # AI generated code display and actions
+                if st.session_state.generated_code:
+                    st.markdown("<div class='card'>", unsafe_allow_html=True)
+                    st.markdown("#### Generated Animation Code")
+                    st.code(st.session_state.generated_code, language="python")
+                    
+                    col_ai1, col_ai2 = st.columns(2)
+                    with col_ai1:
+                        if st.button("Use This Code", key="use_gen_code"):
+                            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"):
+                            with st.spinner("Rendering preview..."):
+                                video_data, status = generate_manim_video(
+                                    st.session_state.generated_code, 
+                                    "mp4", 
+                                    "480p",  # Use lowest quality for preview
+                                    ANIMATION_SPEEDS["Normal"]
+                                )
+                                
+                                if video_data:
+                                    st.video(video_data)
+                                    st.download_button(
+                                        label="Download Preview",
+                                        data=video_data,
+                                        file_name=f"manim_preview_{int(time.time())}.mp4",
+                                        mime="video/mp4"
+                                    )
+                                else:
+                                    st.error(f"Failed to generate preview: {status}")
+                    st.markdown("</div>", unsafe_allow_html=True)
+            else:
+                st.warning("AI models failed to load. Generation features are unavailable.")
         else:
-            st.warning("AI models failed to load. Generation features are unavailable.")
+            st.info("Please enter the correct password to access AI features")
 
     # LATEX FORMULAS TAB
     with tabs[2]: