Update app.py

app.py

@@ -12,7 +12,7 @@ if "messages" not in st.session_state:
 # Display chat messages from history
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
-        st.write(message["content"])
+        st.markdown(message["content"], unsafe_allow_html=True)
 
 # Accept user input
 if prompt := st.chat_input("Ask me about stocks..."):
@@ -30,21 +30,29 @@ if prompt := st.chat_input("Ask me about stocks..."):
         try:
             stock_data = yf.Ticker(company_name).info
             
-            # Check if stock_data contains valid information
-            if 'currentPrice' in stock_data:
-                response = f"Here is the data for {company_name}:\n"
-                response += f"Current Price: {stock_data.get('currentPrice', 'N/A')}\n"
-                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."
+            # Extract relevant information into a structured format
+            stock_info = {
+                "Company": stock_data.get("longName", "N/A"),
+                "Current Price": stock_data.get("currentPrice", "N/A"),
+                "Market Cap": stock_data.get("marketCap", "N/A"),
+                "PE Ratio": stock_data.get("trailingPE", "N/A"),
+                "Dividend Yield": stock_data.get("dividendYield", "N/A"),
+                "52 Week High": stock_data.get("fiftyTwoWeekHigh", "N/A"),
+                "52 Week Low": stock_data.get("fiftyTwoWeekLow", "N/A"),
+                "Sector": stock_data.get("sector", "N/A"),
+                "Industry": stock_data.get("industry", "N/A")
+            }
+
+            # Prepare response string with line breaks for readability
+            response = f"Here is the data for {company_name}:\n"
+            for key, value in stock_info.items():
+                response += f"{key}: {value}\n"
+
+            # Simple investment recommendation logic (this can be improved)
+            if stock_info["PE Ratio"] != "N/A" and float(stock_info["PE Ratio"]) < 20:  # Example condition for recommendation
+                response += "\n**Recommendation:** Yes, consider investing!"
             else:
-                response = f"Sorry, I couldn't find data for {company_name}. Please check the ticker symbol."
+                response += "\n**Recommendation:** No, it might not be a good time to invest."
         
         except Exception as e:
             response = f"An error occurred while fetching data: {str(e)}"
@@ -57,7 +65,7 @@ if prompt := st.chat_input("Ask me about stocks..."):
 
     # Display assistant response in chat message container with line breaks for readability
     with st.chat_message("assistant"):
-        st.write(response.replace("\n", "<br>"), unsafe_allow_html=True)
+        st.markdown(response.replace("\n", "<br>"), unsafe_allow_html=True)
 
-    # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    # Add assistant response to chat history, preserving line breaks
+    st.session_state.messages.append({"role": "assistant", "content": response.replace("\n", "<br>")})