Update app.py

app.py

@@ -1,12 +1,17 @@
 import streamlit as st
 from transformers import pipeline
 
-# Your verified Model ID
+# 1. Page Config MUST be first to tell HF the app is "alive"
+st.set_page_config(page_title="AI Summarizer", page_icon="📝")
+
+st.title("📝 Automated Text Summarizer")
+
+# 2. Configuration
 MODEL_ID = "VoltIC/Automated-Text-Summarizer"
 
 @st.cache_resource
 def load_summarizer():
-    # This worked in your logs!
+    # This function handles the heavy lifting
     return pipeline(
         "summarization", 
         model=MODEL_ID, 
@@ -14,20 +19,33 @@ def load_summarizer():
         framework="pt"
     )
 
-st.set_page_config(page_title="AI TL;DR", page_icon="📝")
-st.title("📝 Automated Text Summarizer")
-
-# Load the model (The logs prove this is working now!)
-summarizer = load_summarizer()
-
-text_input = st.text_area("Paste long text here:", height=300)
+# 3. Deferred Loading Logic
+# This allows the UI to render BEFORE the model finishes downloading
+if 'model_ready' not in st.session_state:
+    st.session_state.model_ready = False
 
-if st.button("Summarize"):
-    if text_input.strip():
-        with st.spinner("AI is thinking..."):
-            # length_penalty=2.0 helps keep it to 2-3 sentences
-            summary = summarizer(text_input, max_length=100, min_length=30, length_penalty=2.0)
-            st.success("Summary Generated!")
-            st.markdown(f"### Result:\n{summary[0]['summary_text']}")
-    else:
-        st.warning("Please enter text first.")
+if not st.session_state.model_ready:
+    with st.status("📥 Downloading AI Model (1.6GB)...", expanded=True) as status:
+        st.write("This only happens on the first run. Please wait 2-3 minutes.")
+        try:
+            summarizer = load_summarizer()
+            st.session_state.model_ready = True
+            status.update(label="✅ Model Loaded!", state="complete", expanded=False)
+            st.rerun()
+        except Exception as e:
+            st.error("Model is still transferring from Hub. Refreshing in 30 seconds...")
+            st.stop()
+else:
+    # 4. The actual interface (only shows once model_ready is True)
+    summarizer = load_summarizer()
+    
+    input_text = st.text_area("Paste your article here:", height=300)
+    
+    if st.button("Summarize"):
+        if input_text.strip():
+            with st.spinner("AI is condensing your text..."):
+                summary = summarizer(input_text, max_length=100, min_length=30)
+                st.subheader("Summary")
+                st.write(summary[0]['summary_text'])
+        else:
+            st.warning("Please enter text first.")