Simplify app startup to fix blank page issue

Replace st.status with st.info/st.spinner for reliability.
Add st.rerun() after initialization so chat input renders
cleanly on second pass.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

app.py

@@ -1,66 +1,18 @@
 """Main Streamlit application for lyric generation."""
 import os
 import streamlit as st
-from src.generator.generator import LyricGenerator
-from config.settings import Settings
 from pathlib import Path
 from datetime import datetime
 
-# Set Hugging Face cache directory to persistent storage
-os.environ['HF_HOME'] = '/data/.huggingface'
-
-# Set SQLite path for local development
-if not Settings.is_huggingface():
+# Set environment before other imports
+if os.getenv('DEPLOYMENT_MODE') == 'huggingface':
+    os.environ['HF_HOME'] = '/data/.huggingface'
+else:
     os.environ['DYLD_LIBRARY_PATH'] = '/usr/local/opt/sqlite/lib'
 
-def initialize_generator():
-    """Initialize the generator with proper error handling"""
-    try:
-        print("\n=== Initializing Generator ===")
-        
-        if Settings.is_huggingface():
-            print("Running in HuggingFace environment")
-            print("Checking environment requirements...")
-            
-            # Debug: List contents of /data directory
-            data_dir = Path("/data")
-            if data_dir.exists():
-                print("\nContents of /data directory:")
-                for item in data_dir.rglob("*"):
-                    if item.is_file():
-                        print(f"- {item.relative_to(data_dir)} "
-                              f"({item.stat().st_size / 1024:.1f} KB)")
-            
-            if not Settings.HF_TOKEN:
-                error_msg = "HuggingFace token not found."
-                print(f"Error: {error_msg}")
-                st.error(error_msg)
-                return None
-            
-            # Ensure persistent storage directory exists
-            storage_path = Path("/data/processed/embeddings")
-            print(f"\nSetting up storage at: {storage_path}")
-            storage_path.mkdir(parents=True, exist_ok=True)
-            
-            if storage_path.exists():
-                print("Storage directory contents:")
-                for item in storage_path.rglob("*"):
-                    if item.is_file():
-                        print(f"- {item.relative_to(storage_path)} "
-                              f"({item.stat().st_size / 1024:.1f} KB)")
-        
-        # Initialize generator
-        print("\nInitializing LyricGenerator...")
-        generator = LyricGenerator()
-        st.success("Generator initialized successfully!")
-        return generator
-        
-    except Exception as e:
-        error_msg = f"Initialization failed: {str(e)}"
-        print(f"\nError: {error_msg}")
-        print(f"Error type: {type(e).__name__}")
-        st.error(error_msg)
-        return None
+from src.generator.generator import LyricGenerator
+from config.settings import Settings
+
 
 def main():
     """Main application function"""
@@ -69,51 +21,36 @@ def main():
         page_icon="🎵",
         layout="wide"
     )
-    
+
     st.title("SongLift LyrGen2")
-    
+
     # Only run startup once per session
     if 'initialized' not in st.session_state:
         print("===== Application Startup at", datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "=====\n")
 
         # Initialize generator
         try:
-            with st.status("Initializing generator...", expanded=True) as status:
-                st.write("Setting up LyricGenerator...")
-                print("\n=== Initializing Generator ===")
-                generator = LyricGenerator()
-                st.session_state.generator = generator
-                st.session_state.chat_history = []
-                st.session_state.current_lyrics = None
-                st.session_state.initialized = True
-                status.update(label="Generator ready!", state="complete")
+            st.info("Initializing generator... this may take a moment.")
+            print("\n=== Initializing Generator ===")
+            generator = LyricGenerator()
+            st.session_state.generator = generator
+            st.session_state.chat_history = []
+            st.session_state.current_lyrics = None
+            st.session_state.initialized = True
+            print("Generator initialized successfully")
+            st.rerun()
         except Exception as e:
             st.error(f"Error initializing generator: {str(e)}")
             print(f"Error: {str(e)}")
             import traceback
             traceback.print_exc()
             st.stop()
-    
-    # Check OpenAI API key at startup
-    if Settings.is_huggingface():
-        print("Running in HuggingFace environment")
-        print("Checking for OpenAI API key in secrets...")
-        if not Settings.OPENAI_API_KEY:
-            st.error(
-                "OpenAI API key not found in HuggingFace secrets. "
-                "Please ensure OPENAI_API_KEY is set in Space settings under Secrets."
-            )
-            print("OpenAI API key not found in HF secrets")
-            st.stop()
-        else:
-            print("OpenAI API key found in HF secrets")
-    else:
-        if not Settings.OPENAI_API_KEY:
-            st.error("OpenAI API key not found in local environment")
-            st.stop()
-    
-    print(f"Current deployment mode: {Settings.DEPLOYMENT_MODE}")
-    
+
+    # Check OpenAI API key
+    if not Settings.OPENAI_API_KEY:
+        st.error("OpenAI API key not found. Please set OPENAI_API_KEY.")
+        st.stop()
+
     # Display chat history
     for message in st.session_state.chat_history:
         user_msg, assistant_msg = message
@@ -121,28 +58,27 @@ def main():
             st.write(user_msg)
         with st.chat_message("assistant"):
             st.markdown(f"```\n{assistant_msg}\n```")
-    
+
     # Chat interface
-    prompt = "Enter your prompt (ask for new lyrics or modify existing ones)..."
-    user_input = st.chat_input(prompt)
-    
+    user_input = st.chat_input("Enter your prompt (ask for new lyrics or modify existing ones)...")
+
     if user_input:
         with st.chat_message("user"):
             st.write(user_input)
-            
+
         with st.chat_message("assistant"):
             try:
-                with st.status("Generating lyrics..."):
+                with st.spinner("Generating lyrics..."):
                     response = st.session_state.generator.generate_lyrics(
                         user_input,
                         st.session_state.chat_history
                     )
-                
+
                 # Store the response
                 lyrics = response['answer']
                 st.markdown(f"```\n{lyrics}\n```")
                 st.session_state.current_lyrics = lyrics
-                
+
                 # Display sources with content
                 with st.expander("View Sources and Context"):
                     st.write("### Top Retrieved Contexts")
@@ -154,11 +90,11 @@ def main():
                         st.write("Content snippet:")
                         st.text(detail['content'])
                         st.write("---")
-                    
+
                     st.write("\n### All Similar Sources")
                     seen_sources = set()
                     unique_sources = []
-                    
+
                     for doc, score in response["source_documents_with_scores"]:
                         source_key = (
                             doc.metadata['artist'],
@@ -167,7 +103,7 @@ def main():
                         if source_key not in seen_sources:
                             seen_sources.add(source_key)
                             unique_sources.append((doc, score))
-                    
+
                     for doc, score in unique_sources:
                         similarity = round((1 - score) * 100, 2)
                         st.write(
@@ -175,12 +111,12 @@ def main():
                             f"{doc.metadata['song_title']} "
                             f"(Similarity: {similarity}%)"
                         )
-                
+
                 # Update chat history
                 st.session_state.chat_history.append((user_input, lyrics))
-                
+
             except Exception as e:
                 st.error(f"Error generating lyrics: {str(e)}")
 
 if __name__ == "__main__":
-    main() 
+    main()