update app

app.py

@@ -1,32 +1,35 @@
-import gradio as gr
+import streamlit as st
 import os
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
-print("🚀 Starting GPT-2 Text Generator...")
+# Set 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") 
 ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
 
-print(f"HF_TOKEN: {'Set' if HF_TOKEN else 'Not set'}")
-print(f"API_KEY: {'Set' if API_KEY else 'Not set'}")
-print(f"ADMIN_PASSWORD: {'Set' if ADMIN_PASSWORD else 'Not set'}")
+@st.cache_resource
+def load_model():
+    """Load and cache the GPT-2 model"""
+    with st.spinner("Loading GPT-2 model..."):
+        try:
+            tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+            model = GPT2LMHeadModel.from_pretrained("gpt2")
+            tokenizer.pad_token = tokenizer.eos_token
+            return tokenizer, model
+        except Exception as e:
+            st.error(f"Error loading model: {e}")
+            return None, None
 
-# Load model and tokenizer
-print("Loading GPT-2 model...")
-try:
-    tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-    model = GPT2LMHeadModel.from_pretrained("gpt2")
-    tokenizer.pad_token = tokenizer.eos_token
-    print("✅ Model loaded successfully!")
-except Exception as e:
-    print(f"❌ Error loading model: {e}")
-    raise e
-
-def generate_text(prompt, max_length=100, temperature=0.7):
-    """Simple text generation function"""
+def generate_text(prompt, max_length, temperature, tokenizer, model):
+    """Generate text using GPT-2"""
     if not prompt:
         return "Please enter a prompt"
     
@@ -34,8 +37,6 @@ def generate_text(prompt, max_length=100, temperature=0.7):
         return "Prompt too long (max 500 characters)"
     
     try:
-        print(f"Generating text for: {prompt[:30]}...")
-        
         # Encode the prompt
         inputs = tokenizer.encode(prompt, return_tensors="pt", max_length=300, truncation=True)
         
@@ -53,79 +54,129 @@ def generate_text(prompt, max_length=100, temperature=0.7):
         
         # Decode the output
         generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        
-        # Extract only the new text (remove the original prompt)
         new_text = generated_text[len(prompt):].strip()
         
-        print(f"✅ Generated {len(new_text)} characters")
         return new_text if new_text else "No text generated. Try a different prompt."
         
     except Exception as e:
-        error_msg = f"Error generating text: {str(e)}"
-        print(f"❌ {error_msg}")
-        return error_msg
+        return f"Error generating text: {str(e)}"
 
-# Create the Gradio interface
-print("Creating Gradio interface...")
+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()
+                else:
+                    st.error("Invalid password")
+            return False
+    return True
 
-with gr.Blocks() as demo:
-    gr.Markdown("# GPT-2 Text Generator")
-    gr.Markdown("Enter a prompt and click generate to create text using GPT-2")
-    
-    with gr.Row():
-        with gr.Column():
-            prompt_input = gr.Textbox(
-                label="Enter your prompt",
-                placeholder="Type your text here...",
-                lines=3
-            )
-            
-            max_length_slider = gr.Slider(
-                minimum=20,
-                maximum=200,
-                value=100,
-                step=10,
-                label="Max length of generated text"
-            )
-            
-            temperature_slider = gr.Slider(
-                minimum=0.1,
-                maximum=1.5,
-                value=0.7,
-                step=0.1,
-                label="Temperature (creativity)"
-            )
-            
-            generate_button = gr.Button("Generate Text", variant="primary")
+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")
+    
+    with col2:
+        if API_KEY:
+            st.success("🔒 API Auth: Enabled")
+        else:
+            st.info("🔒 API Auth: Disabled")
+    
+    with col3:
+        if ADMIN_PASSWORD:
+            st.success("👤 Admin Auth: Active")
+        else:
+            st.info("👤 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
+        )
         
-        with gr.Column():
-            output_text = gr.Textbox(
-                label="Generated Text",
-                lines=8,
-                placeholder="Generated text will appear here..."
-            )
+        # API key input if needed
+        api_key = ""
+        if API_KEY:
+            api_key = st.text_input("API Key:", type="password")
     
-    # Add some example prompts
-    gr.Examples(
-        examples=[
-            "Once upon a time",
-            "The future of technology is",
-            "In a world where",
-        ],
-        inputs=prompt_input
-    )
-    
-    # Connect the generate function
-    generate_button.click(
-        fn=generate_text,
-        inputs=[prompt_input, max_length_slider, temperature_slider],
-        outputs=output_text
-    )
-
-# Launch the app
-print("Launching Gradio app...")
+    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
+    st.subheader("💡 Example Prompts")
+    examples = [
+        "Once upon a time in a distant galaxy,",
+        "The future of artificial intelligence is",
+        "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
+    if hasattr(st.session_state, 'example_prompt'):
+        st.info(f"Example selected: {st.session_state.example_prompt}")
 
 if __name__ == "__main__":
-    demo.launch()
-
-print("✅ App is running!")
+    main()

requirements.txt

@@ -1,4 +1,4 @@
-gradio==4.44.0
+streamlit==1.28.1
 transformers==4.44.2
 torch==2.4.1
 tokenizers==0.19.1