Update app.py

app.py

@@ -32,7 +32,25 @@ from langchain.prompts import HumanMessagePromptTemplate
 from langchain.prompts import ChatMessagePromptTemplate
 from langchain.prompts import ChatPromptTemplate
 
-
+# Function to create a copy-to-clipboard button
+def create_copy_button(text_to_copy):
+    button_uuid = str(uuid.uuid4()).replace("-", "")
+    button_id = re.sub('\D', '', button_uuid)
+    copy_js = f"""
+        <script>
+        function copyToClipboard{button_id}() {{
+            const str = `{text_to_copy}`;
+            const el = document.createElement('textarea');
+            el.value = str;
+            document.body.appendChild(el);
+            el.select();
+            document.execCommand('copy');
+            document.body.removeChild(el);
+        }}
+        </script>
+        <button onclick="copyToClipboard{button_id}()">Copy</button>
+    """
+    return copy_js
 
 # Retrieve the API key from the environment variables
 api_key = os.getenv("OPENAI_API_KEY")
@@ -149,6 +167,10 @@ def main():
                 st.write("Answer:")
                 st.write(ai_response)
 
+    # Create and display the copy button
+    copy_button_html = create_copy_button(ai_response)
+    components.html(copy_button_html, height=30)           
+
      # Add a button to clear chat history
     if st.button("Clear Chat"):
         st.session_state['chat_history'] = []  # Reset the chat history