Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import streamlit as st  
 import google.generativeai as genai  
 from dotenv import load_dotenv  
@@ -10,143 +12,144 @@ load_dotenv()
 api_key = os.getenv("GENERATIVEAI_API_KEY")  
 genai.configure(api_key=api_key)  
   
-# Global variable to maintain chat session  
+# Global variable to maintain chat session and history  
 chat = None  
-  
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []  # Initialize chat history
+if 'input' not in st.session_state:
+    st.session_state.input = ""  # Initialize input state
+
 # Generation configuration and safety settings  
 generation_config = {  
-   "temperature": 0.9,
-"top_p": 0.5,
-"top_k": 5,
-"max_output_tokens": 1000, 
+    "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"  
-   },  
+    {  
+        "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  
+    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:", "")  
+# Display chat history at the top  
+if st.session_state.chat_history:
+    for entry in st.session_state.chat_history:
+        st.markdown(entry, unsafe_allow_html=True)  
+
+# Input area for user text at the bottom  
+user_input = st.text_area("Enter your text here:", value=st.session_state.input)  
   
 # 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.")  
-  
-# Chatbot UI  
-st.write("### Chatbot")  
-st.write("Type a message to get a response from the chatbot.")  
-chat_input = st.text_area("Type a message:", "")  
-if st.button("Send"):  
-   if chat_input.strip():  
-      response = text_summary(chat_input)  
-      st.write("### Response:")  
-      st.markdown(response, unsafe_allow_html=True)  
-   else:  
-      st.warning("Please enter a message to send.")
+    if user_input.strip():  
+        summary = text_summary(user_input, isNew=True)  # Start a new session each time  
+        # Add user input and summary to chat history  
+        st.session_state.chat_history.append(f"<strong>You:</strong><br/>{user_input}<br/><br/>")
+        st.session_state.chat_history.append(f"<strong>Bot:</strong><br/>{summary}<br/><br/>")
+        # Clear the text area by resetting the input session state  
+        st.session_state.input = ""  # Reset input state to empty  
+        st.experimental_rerun()  # Rerun the script to refresh the input field  
+    else:  
+        st.warning("Please enter some text to summarize.")