Update src/streamlit_app.py

src/streamlit_app.py

@@ -77,7 +77,7 @@ with col1:
     st.image(LOGO_URL, width=56)  # fixed width prevents "flash huge"
 with col2:
     st.markdown("## TeapotAI Chat")
-    st.caption("Grounded answers with web context")
+    st.caption("Teapot is a 77 million parameter LLM designed to generate ")
 
 
 # =========================
@@ -89,7 +89,7 @@ with st.sidebar:
     system_prompt = st.text_area(
         "System prompt",
         value=(
-            "You are Teapot, an open-source AI assistant optimized for low-end devices, "
+            "You are Teapot, an open-source AI assistant optimized for running on low-end cpu devices, "
             "providing short, accurate responses without hallucinating while excelling at "
             "information extraction and text summarization. "
             "If the context does not answer the question, reply exactly: "
@@ -238,38 +238,50 @@ for i, msg in enumerate(st.session_state.messages):
             continue
 
         # Assistant
-        st.markdown(msg["content"])
+        
 
         # Info icon popover with full prompt/context
         # (st.popover is stable in your Streamlit range; no rerun on open/close)
         c1, c2 = st.columns([1, 12], vertical_alignment="center")
         with c1:
+            st.markdown(msg["content"])
+            
+        with c2:
+            
+            key = f"fb_{i}"
+            st.session_state.setdefault(key, msg.get("feedback"))
+            st.feedback(
+                "thumbs",
+                key=key,
+                disabled=msg.get("feedback") is not None,
+                on_change=handle_feedback,
+                args=(i,),
+            )
+
+            
+
+        c3, c4 = st.columns([1, 12], vertical_alignment="center")
+        with c3:
+           st.caption(
+                f"🔎 {msg['search_time']:.2f}s  (search)"
+                f"🧠 {msg['gen_time']:.2f}s (generation) "
+                f"⚡ {msg['tps']:.1f} tok/s  "
+                f"🧾 {msg['input_tokens']} input tokens • {msg['output_tokens']} output tokens"
+            )
+            
+        with c4:
             with st.popover("ℹ️", help="Inspect"):
                 st.markdown("**Context**")
                 st.code(msg.get("context", ""), language="text")
-                st.markdown("**System**")
+                st.markdown("**System Prompt**")
                 st.code(msg.get("system_prompt", ""), language="text")
                 st.markdown("**Question**")
                 st.code(msg.get("question", ""), language="text")
-                st.markdown("**Prompt**")
-                st.code(msg.get("prompt", ""), language="text")
-        with c2:
-            st.caption(
-                f"🔎 {msg['search_time']:.2f}s  "
-                f"🧠 {msg['gen_time']:.2f}s  "
-                f"⚡ {msg['tps']:.1f} tok/s  "
-                f"🧾 in {msg['input_tokens']} • out {msg['output_tokens']}"
-            )
+            
 
-        key = f"fb_{i}"
-        st.session_state.setdefault(key, msg.get("feedback"))
-        st.feedback(
-            "thumbs",
-            key=key,
-            disabled=msg.get("feedback") is not None,
-            on_change=handle_feedback,
-            args=(i,),
-        )
+            
+ 
+                
 
 
 # =========================