Update app.py

app.py

@@ -7,13 +7,19 @@ import pickle
 import re
 import os
 
-# Set page config - this must be the first Streamlit command
+# Set Streamlit configuration (instead of using .streamlit/config.toml)
 st.set_page_config(
     page_title="Next Word Predictor",
     page_icon="🔮",
     layout="centered"
 )
 
+# Set other Streamlit configurations
+st.set_option('server.headless', True)
+st.set_option('server.port', 8501)
+st.set_option('server.enableCORS', False)
+st.set_option('server.enableXsrfProtection', False)
+
 # Custom CSS for styling
 st.markdown("""
     <style>
@@ -99,15 +105,16 @@ def main():
     st.title("🔮 Next Word Predictor")
     st.markdown("Enter some text and I'll predict the next word using an LSTM model trained on a large corpus.")
     
+    # Debug: Show files in directory
+    st.sidebar.write("Debug Info:")
+    st.sidebar.write("Files in directory:", os.listdir('.'))
+    
     # Load model and tokenizer
     with st.spinner("Loading model..."):
         model, tokenizer = load_models()
     
     if model is None or tokenizer is None:
         st.error("Failed to load the model. Please check if model files are available.")
-        # Show files in directory for debugging
-        if st.button("Show files in directory"):
-            st.write("Files in directory:", os.listdir('.'))
         return
     
     # Calculate max sequence length (you might want to set this based on your training)