Update src/app.py

src/app.py

@@ -300,24 +300,24 @@ with tab2:
     # The token progress bar will be handled inside the prompt logic based on input length
 
     with c3:
-            # --- NEW FEATURE: DOWNLOAD CHAT ---
-            # Convert history to a readable string
-            chat_log = ""
-            for msg in st.session_state.messages:
-                role = "USER" if msg['role'] == 'user' else "ASSISTANT"
-                chat_log += f"[{role}]: {msg['content']}\n\n"
-                
-            # Only show button if there is history to save
-            if chat_log:
-                st.download_button(
-                    label="💾 Save Chat",
-                    data=chat_log,
-                    file_name="mission_log.txt",
-                    mime="text/plain",
-                    help="Download the current conversation history."
-                )
+        # --- NEW FEATURE: DOWNLOAD CHAT ---
+        # Convert history to a readable string
+        chat_log = ""
+        for msg in st.session_state.messages:
+            role = "USER" if msg['role'] == 'user' else "ASSISTANT"
+            chat_log += f"[{role}]: {msg['content']}\n\n"
+            
+        # Only show button if there is history to save
+        if chat_log:
+            st.download_button(
+                label="💾 Save Chat",
+                data=chat_log,
+                file_name="mission_log.txt",
+                mime="text/plain",
+                help="Download the current conversation history."
+            )
     
-        st.divider()
+    st.divider()
     
     # --- DISPLAY CONVERSATION HISTORY ---
     for message in st.session_state.messages: