Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import openai
+import pyperclip
 
 # Set OpenAI API Key
 openai.api_key = "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"
@@ -49,7 +50,7 @@ def chatbot(user_input, history=[]):
     history.append((f"({topic}) You: {user_input}", f"Motivator Bot: {bot_response}"))
     return history, saved_chats
 
-# Function to display saved chats
+# Gradio Interface setup
 def display_saved_chats():
     display = ""
     for category, chats in saved_chats.items():
@@ -59,7 +60,7 @@ def display_saved_chats():
         display += "</div>"
     return display.strip()
 
-# Gradio Interface setup
+# Chatbot interface with buttons for "Copy Chat" and "Save Screenshot"
 chat_interface = gr.Blocks(css="""
 body {
     font-family: 'Poppins', sans-serif;
@@ -139,21 +140,38 @@ button:hover {
     transform: translateY(-3px);
     box-shadow: 0px 4px 15px rgba(255, 127, 179, 0.5);
 }
+
+h1 {
+    color: #ff758c;
+    font-size: 3rem;
+    text-align: center;
+    animation: colorChange 3s infinite alternate;
+}
+
+@keyframes colorChange {
+    0% { color: #ff758c; }
+    25% { color: #ff7eb3; }
+    50% { color: #ff9a9e; }
+    75% { color: #ffb3d9; }
+    100% { color: #a1c4fd; }
+}
 """)
 
 with chat_interface:
     with gr.Row():
-        gr.Markdown("<h1 style='text-align:center;'>🌈 Vibrant Motivational Chatbot</h1>")
+        gr.Markdown("<h1 style='text-align:center;'>🌈 **MotivaMate** 🌟</h1>")
     with gr.Row():
         gr.Markdown("**Feeling stressed or unmotivated? Share your thoughts and let me help!**")
     with gr.Row():
-        chatbot_output = gr.Chatbot(label="Motivator Bot")
+        chatbot_output = gr.Chatbot(label="Motivator Bot", interactive=True)
     with gr.Row():
         user_input = gr.Textbox(label="Your Message", placeholder="Type something...")
         send_button = gr.Button("Send")
     with gr.Row():
         saved_chats_display = gr.HTML(label="Saved Chats", value=display_saved_chats())
         refresh_button = gr.Button("Refresh Saved Chats")
+        copy_button = gr.Button("Copy Chat")
+        screenshot_button = gr.Button("Save Screenshot")
     
     def handle_interaction(user_input, history):
         if not user_input.strip():
@@ -161,8 +179,18 @@ with chat_interface:
         updated_history, _ = chatbot(user_input, history)
         return updated_history, display_saved_chats()
     
+    def copy_chat_to_clipboard():
+        chat_content = "\n".join([f"You: {user}\nBot: {bot}" for user, bot in saved_chats["General"]])
+        pyperclip.copy(chat_content)  # Copies chat content to clipboard
+        return "Chat copied to clipboard!"
+
+    def save_screenshot():
+        return "Screenshot feature coming soon!"  # Placeholder for future screenshot functionality
+
     send_button.click(handle_interaction, inputs=[user_input, chatbot_output], outputs=[chatbot_output, saved_chats_display])
     refresh_button.click(display_saved_chats, inputs=[], outputs=saved_chats_display)
+    copy_button.click(copy_chat_to_clipboard, inputs=[], outputs=[gr.Text(label="Status"), saved_chats_display])
+    screenshot_button.click(save_screenshot, inputs=[], outputs=[gr.Text(label="Status"), saved_chats_display])
 
 if __name__ == "__main__":
     chat_interface.launch()