update app6

app.py

@@ -3,18 +3,25 @@ import os
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
-# Set page config
+# ----------------------------
+# Page config
+# ----------------------------
 st.set_page_config(
     page_title="GPT-2 Text Generator",
     page_icon="🤖",
     layout="wide"
 )
 
+# ----------------------------
 # Load environment variables
+# ----------------------------
 HF_TOKEN = os.getenv("HF_TOKEN")
-API_KEY = os.getenv("API_KEY") 
+API_KEY = os.getenv("API_KEY")
 ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
 
+# ----------------------------
+# Model loading
+# ----------------------------
 @st.cache_resource
 def load_model():
     """Load and cache the GPT-2 model"""
@@ -28,19 +35,20 @@ def load_model():
             st.error(f"Error loading model: {e}")
             return None, None
 
+# ----------------------------
+# Text generation
+# ----------------------------
 def generate_text(prompt, max_length, temperature, tokenizer, model):
     """Generate text using GPT-2"""
     if not prompt:
         return "Please enter a prompt"
-    
+
     if len(prompt) > 500:
         return "Prompt too long (max 500 characters)"
-    
+
     try:
-        # Encode the prompt
         inputs = tokenizer.encode(prompt, return_tensors="pt", max_length=300, truncation=True)
-        
-        # Generate text
+
         with torch.no_grad():
             outputs = model.generate(
                 inputs,
@@ -51,114 +59,97 @@ def generate_text(prompt, max_length, temperature, tokenizer, model):
                 eos_token_id=tokenizer.eos_token_id,
                 no_repeat_ngram_size=2
             )
-        
-        # Decode the output
+
         generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
         new_text = generated_text[len(prompt):].strip()
-        
+
         return new_text if new_text else "No text generated. Try a different prompt."
-        
+
     except Exception as e:
         return f"Error generating text: {str(e)}"
 
+# ----------------------------
+# Authentication
+# ----------------------------
 def check_auth():
     """Handle authentication"""
     if ADMIN_PASSWORD:
         if "authenticated" not in st.session_state:
             st.session_state.authenticated = False
-        
+
         if not st.session_state.authenticated:
             st.title("🔒 Authentication Required")
             password = st.text_input("Enter admin password:", type="password")
             if st.button("Login"):
                 if password == ADMIN_PASSWORD:
                     st.session_state.authenticated = True
-                    st.rerun()
+                    st.experimental_rerun()
                 else:
                     st.error("Invalid password")
             return False
     return True
 
+# ----------------------------
+# Main UI
+# ----------------------------
 def main():
-    # Check authentication
     if not check_auth():
         return
-    
-    # Load model
+
     tokenizer, model = load_model()
     if tokenizer is None or model is None:
         st.error("Failed to load model. Please check the logs.")
         return
-    
-    # Main interface
+
     st.title("🤖 GPT-2 Text Generator")
     st.markdown("Generate text using GPT-2 language model")
-    
+
     # Security status
     col1, col2, col3 = st.columns(3)
     with col1:
-        if HF_TOKEN:
-            st.success("🔑 HF Token: Active")
-        else:
-            st.info("🔑 HF Token: Not set")
-    
+        st.success("🔑 HF Token: Active" if HF_TOKEN else "🔑 HF Token: Not set")
     with col2:
-        if API_KEY:
-            st.success("🔒 API Auth: Enabled")
-        else:
-            st.info("🔒 API Auth: Disabled")
-    
+        st.success("🔒 API Auth: Enabled" if API_KEY else "🔒 API Auth: Disabled")
     with col3:
-        if ADMIN_PASSWORD:
-            st.success("👤 Admin Auth: Active")
-        else:
-            st.info("👤 Admin Auth: Disabled")
-    
+        st.success("👤 Admin Auth: Active" if ADMIN_PASSWORD else "👤 Admin Auth: Disabled")
+
     # Input section
     st.subheader("📝 Input")
-    
     col1, col2 = st.columns([2, 1])
-    
+
     with col1:
         prompt = st.text_area(
             "Enter your prompt:",
             placeholder="Type your text here...",
             height=100
         )
-        
-        # API key input if needed
         api_key = ""
         if API_KEY:
             api_key = st.text_input("API Key:", type="password")
-    
+
     with col2:
         st.subheader("⚙️ Settings")
         max_length = st.slider("Max Length", 20, 200, 100, 10)
         temperature = st.slider("Temperature", 0.1, 1.5, 0.7, 0.1)
-        
         generate_btn = st.button("🚀 Generate Text", type="primary")
-    
+
     # API key validation
     if API_KEY and generate_btn:
         if not api_key or api_key != API_KEY:
             st.error("🔒 Invalid or missing API key")
             return
-    
+
     # Generate text
     if generate_btn and prompt:
         with st.spinner("Generating text..."):
             result = generate_text(prompt, max_length, temperature, tokenizer, model)
-        
         st.subheader("📄 Generated Text")
         st.text_area("Output:", value=result, height=200)
-        
-        # Copy button
         st.code(result)
-    
     elif generate_btn:
         st.warning("Please enter a prompt")
-    
-    # Examples
+
+    # Example prompts
     st.subheader("💡 Example Prompts")
     examples = [
         "Once upon a time in a distant galaxy,",
@@ -166,17 +157,17 @@ def main():
         "In the heart of the ancient forest,",
         "The detective walked into the room and noticed"
     ]
-    
+
     cols = st.columns(len(examples))
     for i, example in enumerate(examples):
         with cols[i]:
             if st.button(f"Use Example {i+1}", key=f"ex_{i}"):
                 st.session_state.example_prompt = example
-                st.rerun()
-    
-    # Use selected example
+                st.experimental_rerun()
+
     if hasattr(st.session_state, 'example_prompt'):
         st.info(f"Example selected: {st.session_state.example_prompt}")
 
+
 if __name__ == "__main__":
-    main()
+    main()