Update app.py

app.py

@@ -197,10 +197,22 @@ with gr.Blocks(
         color_accent_soft='*secondary_200',
         color_accent_soft_dark='*secondary_800',
         button_primary_background_fill='*secondary_100',
-        button_primary_background_fill_dark='*secondary_600'
+        button_primary_background_fill_dark='*secondary_600', 
+        button_primary_background_fill_dark='*secondary_800',
+        button_primary_text_color='white', 
+        button_primary_border_color='*secondary_700',
+        button_primary_border_color_dark='*secondary_900'
     )
      
 ) as demo:
+        with gr.Row():
+            save_button = gr.Button("💾 Save Chat History", 
+                                  variant="primary",
+                                  size="sm")
+            download_button = gr.File(interactive=False, 
+                                    visible=False,
+                                    elem_classes=["download-btn"])
+            
         gr.ChatInterface(
             fn=respond,
             title="Accessible Intelligence Hub",
@@ -210,7 +222,46 @@ with gr.Blocks(
         with gr.Row(scale=1):
             chatbot_topic=gr.CheckboxGroup(["Academia", "Extracurriculars", "Time Management", "Financial Aid"], label="What would you like advice about?")
 
-
+        demo.css = """
+        .download-btn {
+            min-width: 200px !important;
+        }
+        .download-btn .gr-button {
+            background: var(--button-primary-background-fill) !important;
+            color: var(--button-primary-text-color) !important;
+        }
+        """
+
+        def save_chat_history(history, username):
+            if not username:
+                username = "anonymous"
+            
+            timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+            filename = f"chat_history_{username}_{timestamp}.txt"
+            
+            with open(filename, "w", encoding="utf-8") as f:
+                f.write(f"Chat History for {username} - {timestamp}\n\n")
+                for exchange in history:
+                    if isinstance(exchange, tuple) and len(exchange) == 2:
+                        user_msg, bot_msg = exchange
+                        f.write(f"User: {user_msg}\n")
+                        f.write(f"Bot: {bot_msg}\n\n")
+                    elif isinstance(exchange, dict):
+                        # Handle dictionary format if needed
+                        role = exchange.get("role", "unknown")
+                        content = exchange.get("content", "")
+                        f.write(f"{role.capitalize()}: {content}\n\n")
+            
+            return filename
+
+        save_button.click(
+            fn=save_chat_history,
+            inputs=[chat_interface.chatbot],
+            outputs=download_button
+        ).then(
+            lambda: gr.update(visible=True),
+            outputs=download_button
+        )
 #launching chatbot
 demo.launch()