Update app.py

app.py

@@ -1,11 +1,9 @@
 import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
-from transformers import T5Tokenizer
 import torch
 import os
-import json
 
-# Custom CSS styling
+# Custom CSS styling for a light, elegant design
 st.markdown("""
 <style>
     .main {
@@ -79,12 +77,8 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Load model and tokenizer
-
-
-
-tokenizer = T5Tokenizer.from_pretrained("t5-small")
-tokenizer.save_pretrained("./saved_tokenizer")
-summary_file = "summaries.json"
+model_path = "./saved_model"
+tokenizer_path = "./saved_tokenizer"  # Define this path for saved tokenizer
 
 try:
     if not os.path.exists(tokenizer_path):
@@ -117,51 +111,31 @@ def generate_summary(text):
         st.error(f"Error generating summary: {e}")
         return None
 
-def load_summaries():
-    if os.path.exists(summary_file):
-        with open(summary_file, "r", encoding="utf-8") as f:
-            return json.load(f)
-    return []
-
-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)
+st.title("🧠 Smart Text Summarizer")
 
-# Sidebar Navigation
-page = st.sidebar.selectbox("Choose Page", ["Generate Summary", "Saved Summaries"])
+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)
 
-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.")
+text = st.text_area("Enter the text you want to summarize...", height=200)
 
-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!")
+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.")
 
 st.markdown("""
 <div class="footer">