Update app.py

app.py

@@ -0,0 +1,146 @@
+# app.py
+
+import streamlit as st
+import google.generativeai as genai
+from dotenv import load_dotenv
+import os
+
+# Load environment variables
+load_dotenv()
+
+# Configure Google Generative AI with API key
+api_key = os.getenv("GENERATIVEAI_API_KEY")
+genai.configure(api_key=api_key)
+
+# Global variable to maintain chat session
+chat = None
+
+# Generation configuration and safety settings
+generation_config = {
+    "temperature": 0.9,
+    "top_p": 0.5,
+    "top_k": 5,
+    "max_output_tokens": 1000,
+}
+
+safety_settings = [
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    },
+]
+
+# Function to handle text summary requests
+def text_summary(text, isNew=False):
+    global chat
+
+    if isNew or chat is None:  # Start a new chat session
+        model = genai.GenerativeModel(
+            model_name="gemini-pro",
+            generation_config=generation_config,
+            safety_settings=safety_settings
+        )
+        chat = model.start_chat()
+        chat.send_message("""
+        Act as a financial advisor and generate financial summaries in a structured and tabular format. Follow these guidelines strictly:
+
+        - Start each section with a clear title in <strong> tags.
+        - For key metrics, use a table with two columns: one for the metric name and one for its value.
+        - Use bullet points only for listing risks and growth prospects.
+        - Ensure each section is clearly separated with line breaks.
+        - Do not use bold or italic formatting (, *), except for the specified HTML tags.
+
+        Example format:
+
+        <strong>Company Overview</strong><br/>
+        <p>Company Name: {Company Name}</p>
+        <p>Description: {Company Description}</p>
+        <br/><br/>
+
+        <strong>Stock Performance</strong><br/>
+        <p>Apple Inc. (AAPL) is a highly valued stock...</p>
+        <br/><br/>
+
+        <strong>Key Metrics</strong><br/>
+        <table>
+            <tr>
+                <th>Metric</th>
+                <th>Value</th>
+            </tr>
+            <tr>
+                <td>Market Capitalization</td>
+                <td>$2.7 trillion</td>
+            </tr>
+            <tr>
+                <td>Stock Price</td>
+                <td>$170 per share</td>
+            </tr>
+            <tr>
+                <td>EPS (TTM)</td>
+                <td>$6.15</td>
+            </tr>
+            <tr>
+                <td>P/E Ratio</td>
+                <td>24.34</td>
+            </tr>
+        </table>
+        <br/><br/>
+
+        <strong>Growth Prospects</strong><br/>
+        <ul>
+            <li>iPhone sales growth in emerging markets.</li>
+            <li>Expansion of services revenue.</li>
+            <li>Increased demand for wearable devices.</li>
+            <li>Development of AR/VR technologies.</li>
+        </ul>
+        <br/><br/>
+
+        <strong>Risks</strong><br/>
+        <ul>
+            <li>Competition from other technology companies.</li>
+            <li>Dependence on iPhone sales.</li>
+            <li>Economic downturns.</li>
+            <li>Supply chain disruptions and geopolitical risks.</li>
+        </ul>
+        <br/><br/>
+
+        <strong>Overall</strong><br/>
+        <p>Apple Inc. is a financially strong company with a history of innovation...</p>
+        <br/><br/>
+        """)
+
+    # Send message and return response
+    response = chat.send_message(text)
+    return response.text
+
+# Streamlit UI
+st.title("Financial Summary Chatbot")
+st.write("Welcome to the Financial Summary Chatbot! Enter your text below to get a financial summary.")
+
+# Input area for user text
+user_input = st.text_area("Enter your text here:", "")
+
+# Button to submit the text
+if st.button("Get Summary"):
+    if user_input.strip():
+        summary = text_summary(user_input, isNew=True)  # Start a new session each time
+        st.write("### Summary:")
+        st.markdown(summary, unsafe_allow_html=True)
+    else:
+        st.warning("Please enter some text to summarize.")
+
+if __name__ == '__main__':
+    # Start the Streamlit app
+    st.run()