Update app.py

app.py

@@ -7,16 +7,11 @@ import os
 load_dotenv()
 
 # Configure Google Generative AI with API key
-api_key = os.getenv("GENERATIVEAI_API_KEY")
+api_key = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=api_key)
 
-# Initialize session state for message history if it doesn't exist
-if 'messages' not in st.session_state:
-    st.session_state.messages = []
-
-# Initialize session state for chat
-if 'chat' not in st.session_state:
-    st.session_state.chat = None
+# Global variable to maintain chat session
+chat = None
 
 # Generation configuration and safety settings
 generation_config = {
@@ -45,88 +40,101 @@ safety_settings = [
     },
 ]
 
+# Function to handle text summary requests
 def text_summary(text, isNew=False):
-    if isNew or st.session_state.chat is None:
+    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
         )
-        st.session_state.chat = model.start_chat()
-        st.session_state.chat.send_message("""\
-        Act as a financial advisor and generate financial summaries in a structured and tabular format...
-        """)  # Your existing prompt here
-
-    response = st.session_state.chat.send_message(text)
+        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
 
-# Custom CSS for styling
-st.markdown("""\
-<style>
-body {
-    background-color: #000000;  /* Black background */
-    color: #FFFFFF;              /* White text */
-}
-/* Common styles for the container */
-.container-common {
-    background-color: #333333;  /* Darker gray for containers */
-    border: 1px solid #444444;
-    border-radius: 10px;
-    padding: 20px;              /* Padding for the container */
-    margin: 10px 0;
-    box-shadow: 0 2px 4px rgba(0,0,0,0.2);
-}
-/* Scrollable container */
-.scrollable-container {
-    max-height: 400px;          /* Set a max height for scrolling */
-    overflow-y: auto;           /* Enable vertical scrolling */
-}
-/* Message styling */
-.message-container {
-    background-color: #444444;  /* Dark gray for messages */
-    border-radius: 8px;
-    padding: 10px;              /* Padding for messages */
-    margin: 5px 0;             /* Margin for messages */
-}
-</style>
-""", unsafe_allow_html=True)
-
-# Sidebar for input and button
-with st.sidebar:
-    
-    user_input = st.text_area("", height=150, label_visibility="collapsed")
-    if st.button("Generate Summary", key="send_button"):
-        if user_input.strip():
-            # Get bot response
-            response = text_summary(user_input)
-            # Add to message history
-            st.session_state.messages.append({
-                "input": user_input,
-                "response": response
-            })
-            st.rerun()
-
-
-
-for message in reversed(st.session_state.messages):  # Show newest messages first
-    st.markdown("**Input Text:**")
-    st.text(message["input"][:200] + "..." if len(message["input"]) > 200 else message["input"])
-    st.markdown('</div>', unsafe_allow_html=True)
-
-
-
-# Text area for storing all generated summaries
-
-
-
-# Join all generated summaries into a single string
-all_summaries = "\n\n".join(message["response"] for message in st.session_state.messages)
-
-# Display the summaries in a text area
-st.text_area("Generated Summaries", value=all_summaries, height=300, disabled=True)
-
-st.markdown('</div>', unsafe_allow_html=True)
-
-# Initialize the chat if it's not already initialized
-if st.session_state.chat is None:
-    text_summary("", isNew=True)
+# Streamlit app layout
+st.title("Financial Summary Generator")
+st.write("Enter the text you want summarized and select if you want to start a new session.")
+
+# Input text and options
+input_text = st.text_area("Enter text to summarize:")
+start_new = st.checkbox("Start new session", value=False)
+
+if st.button("Generate Summary"):
+    if input_text:
+        # Generate the summary using the text_summary function
+        result = text_summary(input_text, start_new)
+        st.markdown(result, unsafe_allow_html=True)
+    else:
+        st.write("Please enter some text to summarize.")