Update app.py

app.py

@@ -15,12 +15,19 @@ from langchain_community.document_loaders import TextLoader
 from langchain_text_splitters import CharacterTextSplitter
 import serpapi
 import requests
-
+import streamlit.components.v1 as components
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
 # Initialize logging and load environment variables
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 load_dotenv()
 
+if 'chat_active' not in st.session_state:
+    st.session_state.chat_active = False
+
+
 # Define and validate API keys
 openai_api_key = os.getenv("OPENAI_API_KEY")
 serper_api_key = os.getenv("SERPER_API_KEY")
@@ -31,6 +38,116 @@ if not openai_api_key or not serper_api_key:
 
 openai.api_key = openai_api_key
 
+
+
+
+
+
+EMAIL_ADDRESS = os.getenv("EMAIL_ADDRESS")
+EMAIL_PASSWORD = os.getenv("EMAIL_PASSWORD")
+SMTP_SERVER = "mail.privateemail.com"
+SMTP_PORT = 587 
+st.markdown("""
+    <style>
+    .custom-image img {
+        width: 100px;  /* Set the width to make the image smaller */
+        height: auto;  /* Keep the aspect ratio */
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+def copy_to_clipboard(text):
+    """Creates a button to copy text to clipboard."""
+    escaped_text = text.replace('\n', '\\n').replace('"', '\\"')
+    copy_icon_html = f"""
+    <style>
+    .copy-container {{
+        position: relative;
+        margin-top: 10px;
+        padding-bottom: 30px; /* Space for the button */
+        font-size: 0; /* Hide extra space */
+    }}
+    .copy-button {{
+        background: none;
+        border: none;
+        color: #808080; /* Grey color */
+        cursor: pointer;
+        font-size: 18px; /* Adjust icon size */
+        position: absolute;
+        bottom: 0;
+        right: 0;
+    }}
+    .copy-button:hover {{
+        color: #606060; /* Darker grey on hover */
+    }}
+    .copy-message {{
+        font-size: 12px;
+        color: #4CAF50;
+        margin-left: 10px;
+        display: none;
+    }}
+    </style>
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0-beta3/css/all.min.css">
+    <div class="copy-container">
+        <button class="copy-button" onclick="copyToClipboard()">
+            <i class="fas fa-copy"></i>
+        </button>
+        <span class="copy-message" id="copy_message">Copied!</span>
+    </div>
+    
+    <script>
+    function copyToClipboard() {{
+        var textArea = document.createElement("textarea");
+        textArea.value = "{escaped_text}";
+        document.body.appendChild(textArea);
+        textArea.select();
+        document.execCommand("copy");
+        document.body.removeChild(textArea);
+        
+        var copyMessage = document.getElementById("copy_message");
+        copyMessage.style.display = "inline";
+        setTimeout(function() {{
+            copyMessage.style.display = "none";
+        }}, 2000);
+    }}
+    </script>
+    """
+    components.html(copy_icon_html, height=60)
+
+
+def feedback_email(name, email, feedback):
+    """Sends feedback via email."""
+    recipient_email = "wajahat698@gmail.com"
+    msg = MIMEMultipart()
+    msg['From'] = EMAIL_ADDRESS
+    msg['To'] = recipient_email
+    msg['Subject'] = "Feedback"
+    msg.attach(MIMEText(feedback, 'plain'))
+    
+    try:
+        with smtplib.SMTP(SMTP_SERVER, SMTP_PORT) as server:
+            server.starttls()
+            server.login(EMAIL_ADDRESS, EMAIL_PASSWORD)
+            server.sendmail(EMAIL_ADDRESS, recipient_email, msg.as_string())
+    except Exception as e:
+        st.error(f"Error sending email: {e}")
+
+
+def get_suggestions(response_text):
+    """Generates follow-up questions or suggestions based on the response text."""
+    prompt = f"{response_text}\n\nWhat follow-up questions or suggestions would you give ? List 1-2 only points"
+    response = openai.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": prompt}]
+    )
+    suggestions = response.choices[0].message.content
+    return suggestions
+
+
+
+
+
+
 # Load knowledge base
 @st.cache_resource
 def load_knowledge_base():
@@ -182,7 +299,107 @@ for message in st.session_state.chat_history:
         st.markdown(message["content"])
 
 # Chat input
-if prompt := st.chat_input(""):
+if not st.session_state.chat_active:
+    st.markdown("""
+        <script>
+        document.addEventListener('DOMContentLoaded', (event) => {
+            const svgs = document.querySelectorAll('svg');
+            svgs.forEach(svg => {
+                if (svg.getAttribute('xmlns') === 'http://www.w3.org/2000/svg' && svg.getAttribute('width') === '18' && svg.getAttribute('height') === '18') {
+                    svg.style.display = 'none';
+                }
+            });
+        });
+        </script>
+
+        <style>
+        /* Hide all <a> elements inside elements with block-container and st-emotion-cache-1eo1tir ea3mdgi5 classes */
+        .block-container.st-emotion-cache-1eo1tir.ea3mdgi5 a {
+            display: none !important;
+        }
+
+        /* Ensure links in the sidebar are visible and underlined */
+        .stSidebar a {
+            display: inline !important;
+            text-decoration: underline !important;
+            color: inherit !important;
+        }
+
+        /* Additional styles */
+        .section-container {
+            display: flex;
+            justify-content: center;
+            align-items: stretch;
+            flex-wrap: wrap;
+            gap: 4px;
+        }
+        
+        .section {
+            flex: 1;
+            min-width: 150px;
+            max-width: 90px;
+            min-height: 150px;
+            border: 1px solid #afafaf;
+            border-radius: 10px;
+            padding: 5px;
+            background-color: transparent;
+            margin: 3px;
+            text-align: center;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            box-sizing: border-box;
+            font-size: 12px;
+            transition: background-color 0.3s ease;
+        }
+
+        .section h2 {
+            color: #afafaf;
+            font-size: 14px;
+            margin-bottom: 8px;
+            border-bottom: 1px solid #afafaf;
+            padding-bottom: 4px;
+        }
+
+        .section p {
+            color: #afafaf;
+            font-size: 11px;
+            margin: 5px 0;
+            line-height: 1.4;
+        }
+
+        @media (max-width: 100px) {
+            .section {
+                min-width: 90%;
+                max-width: 90%;
+            }
+        }
+        </style>
+        
+        <h1 style="text-align: center; background: linear-gradient(45deg, #4b25ea, #bd18e6, #fb590e); -webkit-background-clip: text; color: transparent;">How can I help you today?</h1>
+
+        <div class="section-container">
+            <div class="section">
+                <h2>Find</h2>
+                <p>Discover all your great TrustBuilders®. <br> Example: Find Development Trust Builders® for World Vision 
+            </div>
+            <div class="section">
+                <h2>Create</h2>
+                <p>Generate trust-optimised solutions : <br>Example: Find World Vision development TrustBuilders®. Then use them to write a 200-word annual report article. Enthusiastic tone.</p>
+            </div>
+            <div class="section">
+                <h2>Trust-optimise</h2>
+                <p>Paste your LinkedIn profile, EDM or blog and ask TrustAI® to improve it using specific Trust Buckets® and add your specific TrustBuilders® as examples.</p>
+            </div>
+        </div>
+        """, unsafe_allow_html=True)
+
+else:
+    st.write("")
+
+
+
+
+if prompt := st.chat_input("") and not st.session_state.chat_active:
+    st.session_state.chat_active= True
     # Add user message to chat history
     st.session_state.chat_history.append({"role": "user", "content": prompt})
     
@@ -198,7 +415,7 @@ if prompt := st.chat_input(""):
         try:
             # Generate response using the agent executor
             output = agent_executor.invoke({
-                "input": f"{prompt}Be specific with numbers, dates, people, and dollar amount.Search and Provide correct sources of all the results . Respond directly to the query without any introductory phrases or meta-commentary. Your response should be natural and read as if it's addressing the query immediately, without any preamble. Create Creative  Headlines  using main Trust Bucket targeted always use active language like 'Lets dicover'or 'Watch it grow' etc.",
+                "input": f"{prompt}Be specific with numbers, dates, people, and dollar amount.Search and Provide correct sources of all the results . Respond directly to the query without any introductory phrases or meta-commentary. Your response should be natural and read as if it's addressing the query immediately, without any preamble. Create Creative  Headlines  using main Trust Bucket targeted always use active language like 'Lets dicover'or 'Watch it grow' etc. Font and format of the respone should be same donot use any tags <br> etc strictly. ",
 
                 "chat_history": st.session_state.chat_history
             })
@@ -206,10 +423,20 @@ if prompt := st.chat_input(""):
             
             # Display the response
             message_placeholder.markdown(full_response)
+            follow_up_output = agent_executor.invoke({
+                "input": f"Extract 1-2 follow-up questions from the following text: {full_response}",
+                "chat_history": st.session_state.chat_history
+            })
+            follow_up_questions = follow_up_output["output"]
+            if follow_up_questions:
+                st.markdown(f"Follow-up questions:\n{follow_up_questions}")
+
+
         except Exception as e:
             logger.error(f"Error generating response: {e}")
             full_response = "I apologize, but an error occurred while generating the response. Please try again."
             message_placeholder.markdown(full_response)
 
     # Add AI response to chat history
-    st.session_state.chat_history.append({"role": "assistant", "content": full_response})
+    st.session_state.chat_history.append({"role": "assistant", "content": full_response})
+    copy_to_clipboard(full_response)