Update app.py

app.py

@@ -1,17 +1,22 @@
 import streamlit as st
 from transformers import pipeline
-from langdetect import detect
+from datetime import datetime
+import base64
+import os
 
 # Initialize models
 summarizer = pipeline("summarization")
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ur")
+translator_en_ur = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ur")
+translator_ur_en = pipeline("translation", model="Helsinki-NLP/opus-mt-ur-en")
 grammar_corrector = pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
 
+# Set up page config
 st.set_page_config(page_title="CSS AI Assistant by Komal", layout="wide")
 
 st.title("📚 CSS AI Assistant 🇵🇰")
 st.write("Helping students with news summaries, precis, essays & more — for free!")
 
+# Sidebar Menu
 menu = st.sidebar.radio("Select Feature", [
     "Daily News Summary",
     "Precis Evaluation",
@@ -19,6 +24,23 @@ menu = st.sidebar.radio("Select Feature", [
     "Saved Notes"
 ])
 
+# File to save notes
+NOTES_FILE = "saved_notes.txt"
+
+def save_note(title, content):
+    with open(NOTES_FILE, "a", encoding='utf-8') as f:
+        f.write(f"\n## {title}\n{content}\n")
+
+def load_notes():
+    if os.path.exists(NOTES_FILE):
+        with open(NOTES_FILE, "r", encoding='utf-8') as f:
+            return f.read()
+    return "No saved notes yet."
+
+def generate_download_link(text, filename):
+    b64 = base64.b64encode(text.encode()).decode()
+    return f'<a href="data:file/txt;base64,{b64}" download="{filename}">📥 Download Notes</a>'
+
 # --- Daily News Summary ---
 if menu == "Daily News Summary":
     st.header("📰 AI-Powered News Summarizer")
@@ -28,16 +50,42 @@ if menu == "Daily News Summary":
     if st.button("Summarize") and user_news:
         summary = summarizer(user_news, max_length=100, min_length=30, do_sample=False)[0]['summary_text']
         if lang == "Urdu":
-            translation = translator(summary)[0]['translation_text']
+            translated = translator_en_ur(summary)[0]['translation_text']
             st.success("Summary in Urdu:")
-            st.write(translation)
+            st.write(translated)
+            if st.button("Save Urdu Summary"):
+                save_note(f"Urdu Summary - {datetime.now().date()}", translated)
         else:
             st.success("Summary in English:")
             st.write(summary)
+            if st.button("Save English Summary"):
+                save_note(f"English Summary - {datetime.now().date()}", summary)
 
 # --- Precis Evaluation ---
 elif menu == "Precis Evaluation":
     st.header("✍️ Precis Evaluation Tool")
     precis_input = st.text_area("Enter your precis for evaluation:", height=200)
     if st.button("Evaluate Precis") and precis_input:
-        corrected = grammar_corrector("grammar: " + precis_input)[
+        corrected = grammar_corrector("grammar: " + precis_input)[0]['generated_text']
+        st.success("✅ Corrected Precis:")
+        st.write(corrected)
+        if st.button("Save Precis"):
+            save_note(f"Precis - {datetime.now().date()}", corrected)
+
+# --- Essay Feedback ---
+elif menu == "Essay Feedback":
+    st.header("🖋️ Essay Evaluation Tool")
+    essay_input = st.text_area("Paste your essay here:", height=300)
+    if st.button("Evaluate Essay") and essay_input:
+        corrected = grammar_corrector("grammar: " + essay_input)[0]['generated_text']
+        st.success("✅ Suggestions:")
+        st.write(corrected)
+        if st.button("Save Essay"):
+            save_note(f"Essay - {datetime.now().date()}", corrected)
+
+# --- Saved Notes ---
+elif menu == "Saved Notes":
+    st.header("💾 Saved Notes")
+    notes = load_notes()
+    st.markdown(notes)
+    st.markdown(generate_download_link(notes, "css_notes.txt"), unsafe_allow_html=True)