Update app.py

app.py

@@ -10,6 +10,10 @@ load_dotenv()
 api_key = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=api_key)
 
+# Initialize the session state to store chat history
+if 'messages' not in st.session_state:
+    st.session_state['messages'] = []
+
 # Global variable to maintain chat session
 chat = None
 
@@ -53,88 +57,55 @@ def text_summary(text, isNew=False):
         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 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.")
+# Layout for chatbot UI
+st.title("Financial Summary Chatbot")
+
+# Chat history container (This is where the conversation will appear)
+chat_container = st.container()
+
+# Input container (This will stay at the bottom)
+input_container = st.container()
+
+# Function to display the chat history
+def display_chat():
+    with chat_container:
+        # Loop through session messages and display them
+        for message in st.session_state['messages']:
+            if message['role'] == 'user':
+                st.write(f"**You:** {message['content']}")
+            else:
+                st.write(f"**Bot:** {message['content']}")
+
+# Fixed input area at the bottom using the input container
+with input_container:
+    is_new_session = st.checkbox("Start new session", value=False)
+    user_input = st.text_area("Type your message here:", height=100)
+    send_button = st.button("Send")
+
+    # If user presses 'Send'
+    if send_button and user_input:
+        # Store the user's input
+        st.session_state['messages'].append({"role": "user", "content": user_input})
+        
+        # Call the text_summary function to get the bot's response
+        bot_response = text_summary(user_input, is_new_session)
+        
+        # Store the bot's response
+        st.session_state['messages'].append({"role": "bot", "content": bot_response})
+        
+        # Clear the input text area
+        user_input = ""
+
+# Display the chat history
+display_chat()