Update app.py

app.py

@@ -1,8 +1,10 @@
 import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 import torch
+import os
+import json
 
-# Custom CSS styling for a light, elegant design
+# Custom CSS styling
 st.markdown("""
 <style>
     .main {
@@ -78,9 +80,15 @@ st.markdown("""
 # Load model and tokenizer
 model_path = "./saved_model"
 tokenizer_path = "./saved_tokenizer"
+summary_file = "summaries.json"
 
 try:
-    tokenizer = T5Tokenizer.from_pretrained(tokenizer_path, local_files_only=True)
+    if not os.path.exists(tokenizer_path):
+        tokenizer = T5Tokenizer.from_pretrained("t5-small")
+        tokenizer.save_pretrained(tokenizer_path)
+    else:
+        tokenizer = T5Tokenizer.from_pretrained(tokenizer_path, local_files_only=True)
+
     model = T5ForConditionalGeneration.from_pretrained(model_path, local_files_only=True, ignore_mismatched_sizes=True)
     device = torch.device("cpu")
     model.to(device)
@@ -105,34 +113,55 @@ def generate_summary(text):
         st.error(f"Error generating summary: {e}")
         return None
 
-st.title("🧠 Smart Text Summarizer")
+def load_summaries():
+    if os.path.exists(summary_file):
+        with open(summary_file, "r", encoding="utf-8") as f:
+            return json.load(f)
+    return []
 
-st.markdown("""
-<div style="text-align: center; margin-bottom: 30px;">
-    <img src="https://api.placeholder.com/300x150?text=Smart+Summary" width="300" class="header-image">
-</div>
-""", unsafe_allow_html=True)
+def save_summary(original, summary):
+    summaries = load_summaries()
+    summaries.append({"text": original, "summary": summary})
+    with open(summary_file, "w", encoding="utf-8") as f:
+        json.dump(summaries, f, indent=4, ensure_ascii=False)
+
+# Sidebar Navigation
+page = st.sidebar.selectbox("Choose Page", ["Generate Summary", "Saved Summaries"])
 
-text = st.text_area("Enter the text you want to summarize...", height=200)
+if page == "Generate Summary":
+    st.title("🧠 Smart Text Summarizer")
+    text = st.text_area("Enter the text you want to summarize...", height=200)
+    col1, col2, col3 = st.columns([1, 2, 1])
+    with col2:
+        if st.button("🔍 Generate Summary"):
+            if text and model_loaded:
+                with st.spinner("Generating summary..."):
+                    summary = generate_summary(text)
+                    if summary:
+                        st.markdown('<div class="summary-container"><div class="summary-title">📋 Summary</div>' + 
+                                    summary + '</div>', unsafe_allow_html=True)
+                        save_summary(text, summary)
+                    else:
+                        st.error("❌ Failed to generate summary. Please check your input.")
+            elif not model_loaded:
+                st.error("❌ Failed to load model. Please check the application logs.")
+            else:
+                st.warning("⚠️ Please enter text to summarize.")
 
-col1, col2, col3 = st.columns([1, 2, 1])
-with col2:
-    if st.button("🔍 Generate Summary"):
-        if text and model_loaded:
-            with st.spinner("Generating summary..."):
-                summary = generate_summary(text)
-                if summary:
-                    st.markdown('<div class="summary-container"><div class="summary-title">📋 Summary</div>' + 
-                                summary + '</div>', unsafe_allow_html=True)
-                else:
-                    st.error("❌ Failed to generate summary. Please check your input.")
-        elif not model_loaded:
-            st.error("❌ Failed to load model. Please check the application logs.")
-        else:
-            st.warning("⚠️ Please enter text to summarize.")
+elif page == "Saved Summaries":
+    st.title("📚 Saved Summaries")
+    summaries = load_summaries()
+    if summaries:
+        for i, item in enumerate(summaries[::-1], 1):
+            st.markdown(f"<div class='summary-container'><div class='summary-title'>📝 Summary #{i}</div>"
+                        f"<strong>Original:</strong> {item['text']}<br><br>"
+                        f"<strong>Summary:</strong> {item['summary']}</div>", unsafe_allow_html=True)
+    else:
+        st.info("No saved summaries yet. Generate some first!")
 
 st.markdown("""
 <div class="footer">
-    Smart Text Summarizer.
+    Smart Text Summarizer - Crafted with hadheedo
 </div>
 """, unsafe_allow_html=True)
+