Update app.py

app.py

@@ -205,39 +205,34 @@ class DefenseQAChatbot:
             out = self.model(**enc)
             emb = mean_pooling(out.last_hidden_state, enc["attention_mask"])
         return emb[0].cpu().numpy()
-
+    
     def get_response(self, user_prompt, top_k=5, reject=0.55):
         if not user_prompt.strip():
             return "Please ask a question about defense protocols."
-
+    
         q = self._embed_one(user_prompt)
         sims = self.response_embs @ q
         top = np.argpartition(-sims, min(top_k, len(sims)-1))[:top_k]
         top = top[np.argsort(-sims[top])]
         best = top[0]
         score = float(sims[best])
-
+    
+        # إذا الثقة منخفضة جداً
         if score < reject:
-            return f"I couldn't find a reliable answer with sufficient confidence (score: {score:.2f}). Please try rephrasing your question."
-
+            return "I couldn't find a reliable answer. Please try rephrasing your question or ask about specific defense protocols and procedures."
+    
+        # إرجاع الإجابة بدون تقييم
         response_text = self.responses[best]
-        
-        if score >= 0.80:
-            confidence = "🟢 High confidence"
-        elif score >= 0.65:
-            confidence = "🟡 Medium confidence"
-        else:
-            confidence = "🟠 Low confidence"
-        
-        return f"{response_text}\n\n---\n*{confidence} • Score: {score:.2f}*"
+        return response_text
 
+        
 # ========== Password Protection ==========
 def check_password():
     """Returns True if user entered correct password"""
     
     def password_entered():
         """Checks whether password is correct"""
-        # غيّر كلمة السر هنا
+
         CORRECT_PASSWORD = "NWTC@2025"
         
         if st.session_state["password"] == CORRECT_PASSWORD:
@@ -417,6 +412,7 @@ if len(st.session_state.messages) == 0:
         <div style="font-size: 100px; font-weight: 700; color: #475569; margin-bottom: 20px;">
             NWTC
         </div>
+        
         <p style="font-size: 16px; margin: 0; color: #94a3b8;">
             Ask your questions about: C2/STAFF, CA/PA, INTEL/RECON, MEDICAL, OE/ENVIRONMENT, or TACTICS/TECHNIQUES only.
         </p>
@@ -430,12 +426,23 @@ with chat_container:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+
+
             
-# ========== Chat Input ==========
-if prompt := st.chat_input("Write your questions..."):
+# ========== Chat Input & Clear Button ==========
+col1, col2 = st.columns([5, 1])
+
+with col1:
+    prompt = st.chat_input("Enter your defense protocol inquiry here...")
+
+
+with col2:
+    if st.button("🗑️ Clear", use_container_width=True):
+        st.session_state.messages = []
+        st.rerun()
+# Process message
+if prompt:
     st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(prompt)
     
     with st.chat_message("assistant"):
         with st.spinner("Analyzing..."):
@@ -443,13 +450,7 @@ if prompt := st.chat_input("Write your questions..."):
             st.markdown(response)
     
     st.session_state.messages.append({"role": "assistant", "content": response})
-
-# ========== Clear Button ==========
-col1, col2, col3 = st.columns([3, 1, 3])
-with col2:
-    if st.button("🗑️ Clear Chat"):
-        st.session_state.messages = []
-        st.rerun()
+    st.rerun()
 
 # ========== Footer ==========
 st.markdown("""