Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import streamlit as st  
 import google.generativeai as genai  
 from dotenv import load_dotenv  
@@ -12,13 +10,9 @@ load_dotenv()
 api_key = os.getenv("GENERATIVEAI_API_KEY")  
 genai.configure(api_key=api_key)  
   
-# Global variable to maintain chat session and history  
+# Global variable to maintain chat session  
 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,
@@ -133,23 +127,26 @@ def text_summary(text, isNew=False):
 st.title("Financial Summary Chatbot")  
 st.write("Welcome to the Financial Summary Chatbot! Enter your text below to get a financial summary.")  
   
-# 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)  
+# 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  
-        # 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  
+        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.")