Update sessions

app.py

@@ -33,17 +33,29 @@ async def start_chat():
     # Initialize empty chat history for this session
     chat_histories[session_id] = []
     
-    welcome_message = """👋 Welcome to the Stock Price Calculator!
+    welcome_message = """📈 Welcome to the AI Stock Assistant!
 
-I can help you with:
-• Getting real-time stock prices
-• Calculating how many shares you can buy
+I'm your intelligent stock market companion. Here's what I can do:
 
-Try these examples:
-• Type 'AAPL' to get Apple's stock price
-• Ask 'How many MSFT shares can I buy with $10000?'
+1. Get Real-Time Stock Prices 📊
+   • Just type a ticker symbol (e.g., 'AAPL' for Apple)
+   • Or ask naturally (e.g., "What's Microsoft's stock price?")
 
-What would you like to know?"""
+2. Calculate Share Purchases 💰
+   • Ask how many shares you can buy (e.g., "How many TSLA shares for $5000?")
+   • I'll show you the current price and number of shares
+
+3. Smart Features 🧠
+   • I understand company names and ticker symbols
+   • I remember context from previous messages
+   • I support all major stock exchanges
+
+Popular stocks to try:
+• Tech: AAPL (Apple), MSFT (Microsoft), GOOGL (Google)
+• Finance: JPM (JPMorgan), BAC (Bank of America)
+• Retail: WMT (Walmart), COST (Costco)
+
+What would you like to know about the stock market?"""
     
     await cl.Message(content=welcome_message).send()
 
@@ -71,7 +83,9 @@ async def handle_message(message: cl.Message):
         }
         
         # Process the message with agent_node
+        print(f"[Debug] Processing message: {message.content}")
         response = agent_node(state)
+        print(f"[Debug] Agent response: {response}")
         
         # Send the response back to the user
         if isinstance(response, dict) and "output" in response:
@@ -79,13 +93,14 @@ async def handle_message(message: cl.Message):
             history.append({"role": "assistant", "content": response["output"]})
             await cl.Message(content=response["output"]).send()
         else:
+            print(f"[Debug] Invalid response format: {response}")
             await cl.Message(content="Received an invalid response format from the agent.").send()
             
         # Update history in storage
         chat_histories[session_id] = history
             
     except Exception as e:
-        print(f"Error occurred: {e}")
+        print(f"[Debug] Error occurred: {e}")
         await cl.Message(content="Sorry, something went wrong while processing your request.").send()
 
 @cl.on_chat_end

tools.py

@@ -7,18 +7,26 @@ def get_price(ticker):
     try:
         # Clean the ticker symbol
         ticker = ticker.strip().upper()
+        print(f"[Debug] Fetching price for ticker: {ticker}")
+        
         stock = yf.Ticker(ticker)
+        print(f"[Debug] Created yfinance Ticker object for {ticker}")
         
         # Get the stock info
         info = stock.info
+        print(f"[Debug] Retrieved stock info: {info}")
+        
         if 'regularMarketPrice' not in info or info['regularMarketPrice'] is None:
+            print(f"[Debug] No regularMarketPrice found for {ticker}")
             return f"Could not get the current price for {ticker}. Please verify the ticker symbol."
             
         price = info['regularMarketPrice']
         company_name = info.get('longName', ticker)
+        print(f"[Debug] Found price for {company_name} ({ticker}): ${price}")
         
         return f"The current price of {company_name} ({ticker}) is ${price:.2f}"
     except Exception as e:
+        print(f"[Debug] Error in get_price for {ticker}: {str(e)}")
         return f"Error getting price for {ticker}: {str(e)}"
 
 def buying_power_tool(query):