Update app.py

app.py

@@ -30,36 +30,41 @@ if prompt := st.chat_input("Ask me about stocks..."):
         try:
             stock_data = yf.Ticker(company_name).info
             
-            # 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")
-            }
+            # Check if stock_data contains valid information
+            if 'currentPrice' in stock_data and stock_data['currentPrice'] is not None:
+                # 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"
+                # 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!"
+                # 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 += "\n**Recommendation:** No, it might not be a good time to invest."
             else:
-                response += "\n**Recommendation:** No, it might not be a good time to invest."
+                response = f"Sorry, I couldn't find valid data for {company_name}. Please check the ticker symbol."
         
         except Exception as e:
             response = f"An error occurred while fetching data: {str(e)}"
 
     else:
         try:
-            response = llm.invoke(prompt)  # Use the LLM for general questions
+            # Use the LLM for general questions or topics not related to stocks
+            response = llm.invoke(prompt)  
         except Exception as e:
             response = f"An error occurred while processing your request: {str(e)}"