Update app.py

app.py

@@ -18,7 +18,85 @@ from sentence_transformers import SentenceTransformer
 import numpy as np
 
 # ----------------- تنظیمات صفحه -----------------
-st.set_page_config(page_title="چت‌بات ارتش - فقط از PDF", page_icon="🪖", layout="wide")
+st.set_page_config(page_title="چت‌ بات توانا", page_icon="🪖", layout="wide")
+
+st.markdown("""
+    <style>
+    @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@400;700&display=swap');
+    html, body, [class*="css"] {
+        font-family: 'Vazirmatn', Tahoma, sans-serif;
+        direction: rtl;
+        text-align: right;
+    }
+    .stApp {
+        background: url("./military_bg.jpeg") no-repeat center center fixed;
+        background-size: cover;
+        backdrop-filter: blur(2px);
+    }
+    .stChatMessage {
+        background-color: rgba(255,255,255,0.8);
+        border: 1px solid #4e8a3e;
+        border-radius: 12px;
+        padding: 16px;
+        margin-bottom: 15px;
+        box-shadow: 0 4px 10px rgba(0,0,0,0.2);
+        animation: fadeIn 0.4s ease-in-out;
+    }
+    .stTextInput > div > input, .stTextArea textarea {
+        background-color: rgba(255,255,255,0.9) !important;
+        border-radius: 8px !important;
+        direction: rtl;
+        text-align: right;
+        font-family: 'Vazirmatn', Tahoma;
+    }
+    .stButton>button {
+        background-color: #4e8a3e !important;
+        color: white !important;
+        font-weight: bold;
+        border-radius: 10px;
+        padding: 8px 20px;
+        transition: 0.3s;
+    }
+    .stButton>button:hover {
+        background-color: #3c6d30 !important;
+    }
+    .header-text {
+        text-align: center;
+        margin-top: 20px;
+        margin-bottom: 40px;
+        background-color: rgba(255, 255, 255, 0.75);
+        padding: 20px;
+        border-radius: 20px;
+        box-shadow: 0 4px 12px rgba(0,0,0,0.2);
+    }
+    .header-text h1 {
+        font-size: 42px;
+        color: #2c3e50;
+        margin: 0;
+        font-weight: bold;
+    }
+    .subtitle {
+        font-size: 18px;
+        color: #34495e;
+        margin-top: 8px;
+    }
+    @keyframes fadeIn {
+        from { opacity: 0; transform: translateY(10px); }
+        to { opacity: 1; transform: translateY(0); }
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+col1, col2, col3 = st.columns([1, 1, 1])
+with col2:
+    st.image("army.png", width=240)
+
+st.markdown("""
+    <div class="header-text">
+        <h1>چت‌ بات توانا</h1>
+        <div class="subtitle">دستیار هوشمند برای تصمیم‌گیری در میدان نبرد</div>
+    </div>
+""", unsafe_allow_html=True)
 
 # ----------------- بارگذاری مدل FarsiBERT -----------------
 # model_name = "HooshvareLab/bert-fa-zwnj-base"
@@ -44,7 +122,7 @@ def build_pdf_index():
 
         documents = [LangchainDocument(page_content=t) for t in texts]
 
-        sentence_model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
+        sentence_model = SentenceTransformer('HooshvareLab/bert-fa-zwnj-base')
 
         progress_bar = st.progress(0)
         total_docs = len(documents)
@@ -59,6 +137,8 @@ def build_pdf_index():
             embeddings.extend(batch_embeddings)
 
             progress_bar.progress(min((i + batch_size) / total_docs, 1.0))
+                time.sleep(5)
+            progress_bar.empty()
 
         embeddings = np.array(embeddings)
 
@@ -81,7 +161,7 @@ class SimpleRetriever(BaseRetriever):
 
     def _get_relevant_documents(self, query: str) -> List[Document]:
         # فقط از sentence_model استفاده می‌کنیم
-        sentence_model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
+        sentence_model = SentenceTransformer('HooshvareLab/bert-fa-zwnj-base')
         query_embedding = sentence_model.encode(query, convert_to_numpy=True)
 
         similarities = []