Update app.py

app.py

@@ -2,13 +2,12 @@ import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 import torch
 
-# Load model and tokenizer
+# تحميل النموذج والـ Tokenizer
 model_path = "./saved_model"
-tokenizer_path = "./saved_tokenizer"
 
 try:
-    tokenizer = T5Tokenizer.from_pretrained(tokenizer_path)
     model = T5ForConditionalGeneration.from_pretrained(model_path)
+    tokenizer = T5Tokenizer.from_pretrained(model_path)
     device = torch.device("cpu")
     model.to(device)
     model_loaded = True
@@ -16,7 +15,7 @@ except Exception as e:
     st.error(f"Error loading model: {e}")
     model_loaded = False
 
-# Function to generate summary
+# دالة توليد الملخص
 def generate_summary(text):
     try:
         inputs = ["summarize: " + text]
@@ -27,8 +26,40 @@ def generate_summary(text):
         st.error(f"Error generating summary: {e}")
         return None
 
-# Streamlit UI
-st.title("Text Summarization Application")
+# تخصيص الألوان
+primary_color = "#FFB6C1"  # وردي فاتح
+secondary_color = "#E6E6FA"  # بنفسجي فاتح
+text_color = "#4A4A4A"  # رمادي داكن
+
+# تخصيص واجهة المستخدم
+st.set_page_config(page_title="Text Summarizer", page_icon=":sparkles:", layout="wide")
+
+st.markdown(
+    f"""
+    <style>
+        body {{
+            color: {text_color};
+            background-color: {secondary_color};
+        }}
+        .stApp {{
+            background-color: {secondary_color};
+        }}
+        .stButton button {{
+            background-color: {primary_color};
+            color: white;
+        }}
+        .stTextArea textarea {{
+            background-color: white;
+        }}
+        .stTextInput input {{
+            background-color: white;
+        }}
+    </style>
+    """,
+    unsafe_allow_html=True,
+)
+
+st.title("✨ Text Summarization Application ✨")
 
 st.markdown("Enter the text you want to summarize, and the application will generate a concise summary.")
 
@@ -49,4 +80,3 @@ if st.button("Summarize"):
         st.warning("Please enter text.")
 
 st.markdown("---")
-st.markdown("This application was developed using the T5 model.")