Update app.py

app.py

@@ -2,29 +2,8 @@ import streamlit as st
 from langchain_groq import ChatGroq
 import yfinance as yf
 
-# Initialize the ChatGroq model
-llm = ChatGroq(model_name="Llama3-8b-8192", api_key=st.secrets['groq_api_key']) # Ensure this key is set in your Hugging Face secrets
-
-# Custom CSS for dark blue theme
-st.markdown(
-    """
-    <style>
-    .stApp {
-        background-color: #1e1e2f;
-        color: white;
-    }
-    .stTextInput>div>input {
-        background-color: #2e2e3e;
-        color: white;
-    }
-    .stButton>button {
-        background-color: #007bff;
-        color: white;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
+# Initialize the ChatGroq model using the secret API key
+llm = ChatGroq(model_name="Llama3-8b-8192", api_key=st.secrets['groq_api_key'])
 
 # Initialize chat history in session state
 if "messages" not in st.session_state:
@@ -58,6 +37,12 @@ if prompt := st.chat_input("Ask me about stocks..."):
                 response += f"Market Cap: {stock_data.get('marketCap', 'N/A')}\n"
                 response += f"PE Ratio: {stock_data.get('trailingPE', 'N/A')}\n"
                 response += f"Dividend Yield: {stock_data.get('dividendYield', 'N/A')}\n"
+
+                # Simple investment recommendation logic (this can be improved)
+                if stock_data.get('trailingPE', 0) < 20:  # Example condition for recommendation
+                    response += "\n**Recommendation:** Yes, consider investing!"
+                else:
+                    response += "\n**Recommendation:** No, it might not be a good time to invest."
             else:
                 response = f"Sorry, I couldn't find data for {company_name}. Please check the ticker symbol."
         
@@ -70,9 +55,9 @@ if prompt := st.chat_input("Ask me about stocks..."):
         except Exception as e:
             response = f"An error occurred while processing your request: {str(e)}"
 
-    # Display assistant response in chat message container
+    # Display assistant response in chat message container with line breaks for readability
     with st.chat_message("assistant"):
-        st.write(response)
+        st.write(response.replace("\n", "<br>"), unsafe_allow_html=True)
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})