Update app.py

app.py

@@ -45,7 +45,7 @@ if "llama_model" not in st.session_state or "llama_tokenizer" not in st.session_
     st.session_state.llama_model = model
     st.session_state.llama_tokenizer = tokenizer
 
-st.set_page_config(page_title="Ask Wikipedia", page_icon="📘", layout="wide")
+st.set_page_config(page_title="Tell Me Why", page_icon="❓", layout="wide")
 
 def get_llama_response(query):
     # model = st.session_state.llama_model
@@ -69,9 +69,9 @@ def generate_pdf(convo, topic):
     html = f"<h2>{topic}</h2><hr>"
     for msg in convo:
         if msg["role"] == "user":
-            html += f"<p><strong>👤 You:</strong> {msg['text']}</p>"
+            html += f"<p><strong>You:</strong> {msg['text']}</p>"
         elif msg["role"] == "assistant":
-            html += f"<p><strong>📘 Wikipedia:</strong> {msg['text']}</p>"
+            html += f"<p><strong>AI Assistant:</strong> {msg['text']}</p>"
 
     result = io.BytesIO()
     pisa_status = pisa.CreatePDF(io.StringIO(html), dest=result)
@@ -88,7 +88,7 @@ if "edit_mode" not in st.session_state:
     st.session_state.edit_mode = {}
 
 # --- App Title ---
-st.title("💬 Ask Wikipedia - Your Knowledge Companion")
+st.title("💬 Tell Me Why")
 
 # --- Custom CSS ---
 st.markdown("""
@@ -215,7 +215,7 @@ if st.session_state.current_conversation:
                                 try:
                                     new_response = get_llama_response(prompt)
                                 except:
-                                    new_response = "Failed to retrieve summary."
+                                    new_response = "Failed to retrieve response."
                             if idx + 1 < len(convo) and convo[idx + 1]["role"] == "assistant":
                                 convo[idx + 1]["text"] = new_response
                             st.session_state.edit_mode[idx] = False
@@ -234,7 +234,7 @@ if st.session_state.current_conversation:
                     with col2:
                         st.markdown(f'''
                         <div class="user-bubble">
-                            <div class="bubble-header">👤 <span>User</span></div>
+                            <div class="bubble-header">👤 <span>You</span></div>
                             {msg["text"]}
                         </div>
                         ''', unsafe_allow_html=True)
@@ -242,7 +242,7 @@ if st.session_state.current_conversation:
             elif msg["role"] == "assistant":
                 st.markdown(f'''
                 <div class="assistant-bubble">
-                    <div class="bubble-header">📘 <span>Wikipedia</span></div>
+                    <div class="bubble-header">📘 <span>AI Assistant</span></div>
                     {msg["text"]}
                 </div>
                 ''', unsafe_allow_html=True)