Update app2.py

app2.py

@@ -7,16 +7,15 @@ 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
+# Initialize the session state to store chat history
 if 'messages' not in st.session_state:
-    st.session_state.messages = []
+    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,109 +44,68 @@ 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
+        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.
+        """)
 
-    response = st.session_state.chat.send_message(text)
+    # Send message and return response
+    response = chat.send_message(text)
     return response.text
 
-# Custom CSS for chat interface
-st.markdown("""
-<style>
-.chat-container {
-    height: 600px;
-    overflow-y: auto;
-    padding: 20px;
-    margin-bottom: 100px;
-}
-.user-message {
-    background-color: #e6f3ff;
-    padding: 10px;
-    border-radius: 10px;
-    margin: 10px 0;
-    max-width: 80%;
-    margin-left: auto;
-}
-.bot-message {
-    background-color: #f0f0f0;
-    padding: 10px;
-    border-radius: 10px;
-    margin: 10px 0;
-    max-width: 80%;
-}
-.input-container {
-    position: fixed;
-    bottom: 0;
-    left: 0;
-    right: 0;
-    padding: 20px;
-    background-color: white;
-    border-top: 1px solid #ddd;
-}
-.stTextInput input {
-    border-radius: 20px;
-}
-</style>
-""", unsafe_allow_html=True)
-
-# Main title
+# Layout for chatbot UI
 st.title("Financial Summary Chatbot")
 
-# Chat message container
+# Chat history container (This is where the conversation will appear)
 chat_container = st.container()
 
-# Function to process the message
-def process_message(message: str):
-    if message.strip():
-        # Add user message to history
-        st.session_state.messages.append({"role": "user", "content": message})
-        
-        # Get bot response
-        response = text_summary(message)
-        
-        # Add bot response to history
-        st.session_state.messages.append({"role": "assistant", "content": response})
+# Input container (This will stay at the bottom)
+input_container = st.container()
 
-# Create a container for the input area at the bottom
-with st.container():
-    st.markdown('<div class="input-container">', unsafe_allow_html=True)
-    
-    # Create two columns for input and button
-    col1, col2 = st.columns([5,1])
-    
-    with col1:
-        user_input = st.text_input("Message", key="user_input", label_visibility="collapsed")
-    
-    with col2:
-        if st.button("Send"):
-            process_message(user_input)
-            # Instead of trying to clear the input directly, we'll use a rerun
-            st.rerun()
-    
-    st.markdown('</div>', unsafe_allow_html=True)
+# 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']}")
 
-# Handle Enter key press
-if user_input and len(user_input.strip()) > 0:
-    if '\n' in user_input or st.session_state.get('enter_pressed', False):
-        process_message(user_input)
-        st.session_state.enter_pressed = False
-        st.rerun()
+# 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 chat messages in the container
-with chat_container:
-    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
-    for message in st.session_state.messages:
-        if message["role"] == "user":
-            st.markdown(f'<div class="user-message">{message["content"]}</div>', unsafe_allow_html=True)
-        else:
-            st.markdown(f'<div class="bot-message">{message["content"]}</div>', unsafe_allow_html=True)
-    st.markdown('</div>', unsafe_allow_html=True)
+# Display the chat history
+display_chat()