Update app.py

app.py

@@ -8,18 +8,17 @@ from transformers import pipeline
 st.set_page_config(page_title="AI Reading Assistant", layout="wide")
 
 st.title("📚 AI Reading Assistant")
-st.write("Upload a PDF or paste text. Click a word/sentence to get explanation, meaning, and simplification.")
+st.write("Upload a PDF or paste text. Get simple explanations, summaries, and answers.")
 
 # -----------------------------
-# LOAD MODELS (FREE HF MODELS)
+# LOAD MODEL (LIGHT + STABLE)
 # -----------------------------
 @st.cache_resource
-def load_models():
-    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    qa_model = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
-    return summarizer, qa_model
+def load_model():
+    generator = pipeline("text2text-generation", model="google/flan-t5-base")
+    return generator
 
-summarizer, qa_model = load_models()
+generator = load_model()
 
 # -----------------------------
 # FILE UPLOAD
@@ -31,10 +30,11 @@ text_data = ""
 if uploaded_file:
     reader = PdfReader(uploaded_file)
     for page in reader.pages:
-        text_data += page.extract_text()
+        if page.extract_text():
+            text_data += page.extract_text()
 
 # -----------------------------
-# TEXT INPUT OPTION
+# TEXT INPUT
 # -----------------------------
 text_input = st.text_area("Or paste your text here:")
 
@@ -42,42 +42,54 @@ if text_input:
     text_data = text_input
 
 # -----------------------------
-# DISPLAY TEXT
+# MAIN FUNCTIONALITY
 # -----------------------------
 if text_data:
-    st.subheader("📄 Your Text")
-    st.write(text_data[:2000])  # limit display for performance
+
+    st.subheader("📄 Your Text Preview")
+    st.write(text_data[:1500])
 
     # -----------------------------
     # SIMPLIFY TEXT
     # -----------------------------
     if st.button("✨ Simplify Paragraph"):
         with st.spinner("Simplifying..."):
-            summary = summarizer(text_data[:1000], max_length=120, min_length=30, do_sample=False)
-            st.success(summary[0]['summary_text'])
+            prompt = f"Explain this in very simple English:\n{text_data[:500]}"
+            response = generator(prompt, max_length=150)
+            st.success(response[0]['generated_text'])
+
+    # -----------------------------
+    # SUMMARIZE TEXT
+    # -----------------------------
+    if st.button("📝 Summarize Text"):
+        with st.spinner("Summarizing..."):
+            prompt = f"Summarize this text:\n{text_data[:500]}"
+            response = generator(prompt, max_length=120)
+            st.success(response[0]['generated_text'])
 
     # -----------------------------
     # QUESTION ANSWERING
     # -----------------------------
-    st.subheader("❓ Ask a Question from Text")
+    st.subheader("❓ Ask a Question")
     question = st.text_input("Enter your question:")
 
     if question:
-        with st.spinner("Finding answer..."):
-            answer = qa_model(question=question, context=text_data[:2000])
-            st.success(f"Answer: {answer['answer']}")
+        with st.spinner("Thinking..."):
+            prompt = f"Answer the question based on the text below:\n\nText:\n{text_data[:700]}\n\nQuestion:\n{question}"
+            response = generator(prompt, max_length=120)
+            st.success(response[0]['generated_text'])
 
     # -----------------------------
-    # WORD MEANING (SIMPLE)
+    # WORD EXPLANATION
     # -----------------------------
     st.subheader("🔍 Word Explanation")
     word = st.text_input("Enter a difficult word:")
 
     if word:
-        explanation_prompt = f"Explain the meaning of the word '{word}' in simple English with an example."
         with st.spinner("Explaining..."):
-            explanation = summarizer(explanation_prompt, max_length=60, min_length=20, do_sample=False)
-            st.success(explanation[0]['summary_text'])
+            prompt = f"Explain the word '{word}' in simple English and give an example."
+            response = generator(prompt, max_length=80)
+            st.success(response[0]['generated_text'])
 
 else:
     st.info("Upload a PDF or paste text to begin.")
@@ -86,4 +98,4 @@ else:
 # FOOTER
 # -----------------------------
 st.markdown("---")
-st.caption("Built with ❤️ using Streamlit + Hugging Face")
+st.caption("Built with ❤️ using Streamlit + Hugging Face (FLAN-T5)")