fix: Restore full app functionality

app.py

@@ -1,4 +1,112 @@
 """Main Streamlit application for lyric generation."""
+import os
 import streamlit as st
+from src.generator.generator import LyricGenerator
+from config.settings import Settings
 
-st.write("Hello World!") 
+# Set SQLite path for local development
+if not Settings.is_huggingface():
+    os.environ['DYLD_LIBRARY_PATH'] = '/usr/local/opt/sqlite/lib'
+
+def initialize_generator():
+    """Initialize the generator with proper error handling"""
+    try:
+        # Initialize generator
+        st.info("Loading embeddings...")
+        generator = LyricGenerator()
+        st.success("Embeddings loaded successfully!")
+        return generator
+        
+    except Exception as e:
+        st.error(f"Failed to initialize generator: {str(e)}")
+        return None
+
+def main():
+    """Main application function"""
+    st.set_page_config(
+        page_title="SongLift LyrGen",
+        page_icon="🎵",
+        layout="wide"
+    )
+    
+    st.title("SongLift LyrGen")
+    
+    # Initialize generator on first run
+    if 'generator' not in st.session_state:
+        generator = initialize_generator()
+        if generator is None:
+            st.stop()
+        st.session_state.generator = generator
+        st.session_state.chat_history = []
+        st.session_state.current_lyrics = None
+    
+    # Display chat history
+    for message in st.session_state.chat_history:
+        user_msg, assistant_msg = message
+        with st.chat_message("user"):
+            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)
+    
+    if user_input:
+        with st.chat_message("user"):
+            st.write(user_input)
+            
+        with st.chat_message("assistant"):
+            try:
+                with st.status("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")
+                    for detail in response["context_details"]:
+                        st.write(
+                            f"\n**{detail['artist']} - {detail['song']}** "
+                            f"(Similarity: {detail['similarity']}%)"
+                        )
+                        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'],
+                            doc.metadata['song_title']
+                        )
+                        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(
+                            f"- {doc.metadata['artist']} - "
+                            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()