Update app.py

app.py

@@ -9,13 +9,17 @@ import json
 import asyncio
 import aiohttp
 from typing import Optional, List, Dict
-import nest_asyncio
+import os
 
 # Apply nest_asyncio for async operations in Streamlit
-nest_asyncio.apply()
+try:
+    import nest_asyncio
+    nest_asyncio.apply()
+except ImportError:
+    pass
 
-# Hugging Face token setup
-HUGGINGFACE_TOKEN = st.secrets.get("HUGGINGFACE_TOKEN", "")
+# Get Hugging Face token from environment variables (set in HF Spaces secrets)
+HUGGINGFACE_TOKEN = os.environ.get("HUGGINGFACE_TOKEN", "")
 
 if HUGGINGFACE_TOKEN:
     try:
@@ -24,64 +28,90 @@ if HUGGINGFACE_TOKEN:
     except Exception as e:
         st.error(f"❌ Hugging Face login failed: {e}")
 
-# Custom CSS for silver/sky blue theme
+# Custom CSS for beautiful silver/sky blue theme
 st.markdown("""
 <style>
     .main {
-        background-color: #f8f9fa;
+        background: linear-gradient(135deg, #f8f9fa 0%, #e3f2fd 100%);
+        font-family: 'Inter', sans-serif;
     }
+    
     .stChatMessage {
-        padding: 16px;
-        border-radius: 12px;
-        margin: 8px 0;
-    }
-    .stChatMessage[data-testid="stChatMessage"] {
-        border: 1px solid #e1e5e9;
+        padding: 20px;
+        border-radius: 16px;
+        margin: 12px 0;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.1);
+        border: none;
     }
+    
     /* User message styling */
     .stChatMessage:has(div[data-testid="stChatMessageContent"] > div:first-child > div:first-child) {
-        background-color: #e3f2fd;
-        border-left: 4px solid #2196f3;
+        background: linear-gradient(135deg, #e3f2fd 0%, #bbdefb 100%);
+        border-left: 6px solid #2196f3;
     }
+    
     /* Assistant message styling */
     .stChatMessage:has(div[data-testid="stChatMessageContent"] > div:first-child > div:last-child) {
-        background-color: #ffffff;
-        border-left: 4px solid #78909c;
+        background: linear-gradient(135deg, #ffffff 0%, #f5f5f5 100%);
+        border-left: 6px solid #78909c;
+        border: 1px solid #e0e0e0;
     }
+    
     .stTextInput > div > div > input {
         border: 2px solid #90caf9;
-        border-radius: 20px;
-        padding: 12px 16px;
+        border-radius: 25px;
+        padding: 15px 20px;
+        font-size: 16px;
+        background: white;
+        box-shadow: 0 2px 10px rgba(33, 150, 243, 0.1);
     }
+    
     .stButton > button {
-        background-color: #2196f3;
+        background: linear-gradient(135deg, #2196f3 0%, #1976d2 100%);
         color: white;
         border: none;
-        border-radius: 20px;
-        padding: 10px 24px;
-        font-weight: 500;
+        border-radius: 25px;
+        padding: 12px 28px;
+        font-weight: 600;
+        font-size: 16px;
+        box-shadow: 0 4px 12px rgba(33, 150, 243, 0.3);
+        transition: all 0.3s ease;
     }
+    
     .stButton > button:hover {
-        background-color: #1976d2;
-        color: white;
+        transform: translateY(-2px);
+        box-shadow: 0 6px 16px rgba(33, 150, 243, 0.4);
     }
+    
     .sidebar .sidebar-content {
-        background-color: #eceff1;
+        background: linear-gradient(180deg, #eceff1 0%, #cfd8dc 100%);
+        border-right: 1px solid #b0bec5;
     }
+    
     .api-status {
-        padding: 8px;
-        border-radius: 6px;
-        margin: 5px 0;
-        font-size: 0.8em;
+        padding: 12px;
+        border-radius: 10px;
+        margin: 8px 0;
+        font-size: 14px;
+        font-weight: 500;
     }
+    
     .api-success {
-        background-color: #e8f5e8;
+        background: linear-gradient(135deg, #e8f5e8 0%, #c8e6c9 100%);
         border-left: 4px solid #4caf50;
     }
+    
     .api-error {
-        background-color: #ffebee;
+        background: linear-gradient(135deg, #ffebee 0%, #ffcdd2 100%);
         border-left: 4px solid #f44336;
     }
+    
+    .header-gradient {
+        background: linear-gradient(135deg, #2196f3 0%, #1976d2 100%);
+        -webkit-background-clip: text;
+        -webkit-text-fill-color: transparent;
+        background-clip: text;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -105,7 +135,7 @@ COMPANY_INFO = {
     "stage": "Growth and development, focused on building sustainable processes and systems"
 }
 
-# API Configuration
+# API Configuration with environment variables
 API_CONFIG = {
     "news_api": {
         "url": "https://newsapi.org/v2/top-headlines",
@@ -113,14 +143,14 @@ API_CONFIG = {
             "country": "us",
             "category": "technology",
             "pageSize": 5,
-            "apiKey": st.secrets.get("NEWS_API_KEY", "demo_key")
+            "apiKey": os.environ.get("NEWS_API_KEY", "demo_key")
         }
     },
     "weather_api": {
         "url": "https://api.openweathermap.org/data/2.5/weather",
         "params": {
             "q": "Lagos,NG",
-            "appid": st.secrets.get("WEATHER_API_KEY", "demo_key"),
+            "appid": os.environ.get("WEATHER_API_KEY", "demo_key"),
             "units": "metric"
         }
     },
@@ -129,11 +159,8 @@ API_CONFIG = {
         "params": {
             "function": "GLOBAL_QUOTE",
             "symbol": "AAPL",
-            "apikey": st.secrets.get("ALPHA_VANTAGE_KEY", "demo_key")
+            "apikey": os.environ.get("ALPHA_VANTAGE_KEY", "demo_key")
         }
-    },
-    "yah_tech_internal": {
-        "url": "http://localhost:8000"  # For internal FastAPI server
     }
 }
 
@@ -151,7 +178,7 @@ class APIClient:
     async def make_async_request(self, url: str, params: dict = None, headers: dict = None):
         """Make asynchronous API request"""
         try:
-            async with self.session.get(url, params=params, headers=headers) as response:
+            async with self.session.get(url, params=params, headers=headers, timeout=10) as response:
                 if response.status == 200:
                     return await response.json()
                 else:
@@ -181,9 +208,9 @@ class ExternalAPIService:
         if config["params"]["apiKey"] == "demo_key":
             return {
                 "articles": [
-                    {"title": "AI Technology Advances in 2024", "description": "Latest breakthroughs in artificial intelligence..."},
-                    {"title": "Venture Studio Model Gains Popularity", "description": "More entrepreneurs adopting the venture studio approach..."},
-                    {"title": "Economic Impact of Tech Startups", "description": "How technology startups are reshaping economies..."}
+                    {"title": "🚀 AI Technology Advances in 2024", "description": "Latest breakthroughs in artificial intelligence and machine learning transforming industries..."},
+                    {"title": "💼 Venture Studio Model Gains Popularity", "description": "More entrepreneurs adopting the venture studio approach for scalable business creation..."},
+                    {"title": "🌍 Economic Impact of Tech Startups", "description": "How technology startups are reshaping global economies and creating new opportunities..."}
                 ]
             }
         
@@ -197,9 +224,10 @@ class ExternalAPIService:
         
         if config["params"]["appid"] == "demo_key":
             return {
-                "weather": [{"description": "clear sky"}],
-                "main": {"temp": 28, "humidity": 65},
-                "name": city
+                "weather": [{"description": "clear sky", "main": "Clear"}],
+                "main": {"temp": 28, "humidity": 65, "feels_like": 30},
+                "name": city,
+                "sys": {"country": "NG"}
             }
         
         result = await self.api_client.make_async_request(config["url"], config["params"])
@@ -213,6 +241,7 @@ class ExternalAPIService:
         if config["params"]["apikey"] == "demo_key":
             return {
                 "Global Quote": {
+                    "01. symbol": symbol,
                     "05. price": "185.50",
                     "09. change": "+1.25",
                     "10. change percent": "+0.68%"
@@ -223,7 +252,8 @@ class ExternalAPIService:
         return result
     
     def get_crypto_data(self, coin: str = "bitcoin"):
-        """Get cryptocurrency data (synchronous for simplicity)"""
+        """Get cryptocurrency data"""
+        # Using CoinGecko API (no key required for basic usage)
         url = f"https://api.coingecko.com/api/v3/simple/price"
         params = {
             "ids": coin,
@@ -234,27 +264,27 @@ class ExternalAPIService:
         result = self.api_client.make_sync_request(url, params)
         return result
 
-# Knowledge base for the AI (updated with API capabilities)
+# Knowledge base for the AI
 KNOWLEDGE_BASE = {
     "personal": {
-        "who are you": f"I'm {PERSONAL_INFO['name']}, {PERSONAL_INFO['role']}.",
-        "what are your goals": f"My goals include: {', '.join(PERSONAL_INFO['goals'])}.",
-        "what do you do": f"I'm the CEO of YAH Tech, focused on {PERSONAL_INFO['goals'][2]}.",
-        "tell me about yourself": f"I'm {PERSONAL_INFO['name']}, {PERSONAL_INFO['role']}. My mission is to {PERSONAL_INFO['goals'][0].lower()} and {PERSONAL_INFO['goals'][1].lower()}.",
+        "who are you": f"🤵 I'm **{PERSONAL_INFO['name']}**, {PERSONAL_INFO['role']}.",
+        "what are your goals": f"🎯 My goals include:\n• {PERSONAL_INFO['goals'][0]}\n• {PERSONAL_INFO['goals'][1]}\n• {PERSONAL_INFO['goals'][2]}",
+        "what do you do": f"💼 I'm the CEO of YAH Tech, focused on **{PERSONAL_INFO['goals'][2]}**.",
+        "tell me about yourself": f"🌟 I'm **{PERSONAL_INFO['name']}**, {PERSONAL_INFO['role']}. My mission is to **{PERSONAL_INFO['goals'][0].lower()}** and **{PERSONAL_INFO['goals'][1].lower()}** through innovative technology solutions.",
     },
     "company": {
-        "what is yah tech": f"{COMPANY_INFO['name']} is a {COMPANY_INFO['type']}. Our purpose is to {COMPANY_INFO['purpose'].lower()}.",
-        "what does yah tech do": f"We specialize in {COMPANY_INFO['primary_activity'].lower()}. {COMPANY_INFO['purpose']}.",
-        "company philosophy": f"Our philosophy is: '{COMPANY_INFO['philosophy']}'.",
-        "company stage": f"We're currently in the {COMPANY_INFO['stage']} phase.",
-        "venture studio": "As a venture studio, we don't just build apps - we build complete businesses with scalable systems and futuristic solutions.",
+        "what is yah tech": f"🏢 **{COMPANY_INFO['name']}** is a {COMPANY_INFO['type']}. Our purpose is to **{COMPANY_INFO['purpose'].lower()}**.",
+        "what does yah tech do": f"🚀 We specialize in **{COMPANY_INFO['primary_activity'].lower()}**. {COMPANY_INFO['purpose']}.",
+        "company philosophy": f"💡 Our philosophy is: **'{COMPANY_INFO['philosophy']}'**",
+        "company stage": f"📈 We're currently in the **{COMPANY_INFO['stage']}** phase.",
+        "venture studio": "🎯 As a venture studio, we don't just build apps - we build complete businesses with scalable systems and futuristic solutions that drive economic growth.",
     },
     "api_capabilities": {
-        "what can you do": "I can fetch real-time data like tech news, weather, stock prices, and cryptocurrency data. I also have knowledge about YAH Tech and Adedoyin James.",
-        "get news": "I can fetch the latest technology news for you.",
-        "weather": "I can check weather information for any city.",
-        "stock": "I can get stock market data for companies.",
-        "crypto": "I can fetch cryptocurrency prices and changes.",
+        "what can you do": "🔧 I can fetch real-time data like:\n• 📰 Latest tech news\n• 🌤️ Weather information\n• 📈 Stock market data\n• ₿ Cryptocurrency prices\n\nI also have extensive knowledge about YAH Tech and Adedoyin James!",
+        "get news": "📰 I can fetch the latest technology news to keep you updated with industry trends.",
+        "weather": "🌤️ I can check weather information for any city worldwide.",
+        "stock": "📈 I can get real-time stock market data for major companies.",
+        "crypto": "₿ I can fetch current cryptocurrency prices and market changes.",
     }
 }
 
@@ -269,7 +299,7 @@ class YAHBot:
     def _load_model(self):
         """Load the Hugging Face model"""
         try:
-            with st.spinner("🔄 Loading AI model..."):
+            with st.spinner("🔄 Loading YAH Tech AI model..."):
                 self.tokenizer = AutoTokenizer.from_pretrained(
                     "Adedoyinjames/YAH-Tech-Chat-Bot", 
                     use_auth_token=HUGGINGFACE_TOKEN if HUGGINGFACE_TOKEN else True
@@ -279,16 +309,16 @@ class YAHBot:
                     use_auth_token=HUGGINGFACE_TOKEN if HUGGINGFACE_TOKEN else True,
                     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
                 )
-            st.success("✅ AI model loaded successfully!")
+            st.success("✅ YAH Tech AI model loaded successfully!")
         except Exception as e:
             st.error(f"❌ Failed to load custom model: {e}")
             st.info("🔄 Falling back to standard FLAN-T5-small model...")
             try:
                 self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
                 self.model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
-                st.success("✅ Standard model loaded successfully!")
+                st.success("✅ Standard AI model loaded successfully!")
             except Exception as e2:
-                st.error(f"❌ Failed to load standard model: {e2}")
+                st.error(f"❌ Failed to load AI model: {e2}")
     
     def _get_knowledge_response(self, user_input):
         """Check if the input matches any knowledge base entries"""
@@ -316,29 +346,29 @@ class YAHBot:
         if any(word in user_input_lower for word in ["news", "headlines", "tech news"]):
             news_data = await self.api_service.get_tech_news()
             if "articles" in news_data:
-                articles = news_data["articles"][:3]  # Show top 3
-                response = "📰 **Latest Tech News:**\n\n"
+                articles = news_data["articles"][:3]
+                response = "📰 **Latest Technology News:**\n\n"
                 for i, article in enumerate(articles, 1):
-                    response += f"{i}. **{article.get('title', 'No title')}**\n"
+                    response += f"**{i}. {article.get('title', 'No title')}**\n"
                     response += f"   {article.get('description', 'No description available')}\n\n"
                 return response
             else:
                 return "❌ Could not fetch news at the moment. Please try again later."
         
         elif any(word in user_input_lower for word in ["weather", "temperature"]):
-            # Extract city name if mentioned
             city = "Lagos"
             for word in user_input_lower.split():
-                if word not in ["weather", "temperature", "what's", "the"] and len(word) > 2:
+                if word not in ["weather", "temperature", "what's", "the", "in", "for"] and len(word) > 2:
                     city = word.capitalize()
                     break
             
             weather_data = await self.api_service.get_weather(city)
             if "weather" in weather_data:
                 temp = weather_data["main"]["temp"]
-                desc = weather_data["weather"][0]["description"]
+                desc = weather_data["weather"][0]["description"].title()
                 humidity = weather_data["main"]["humidity"]
-                return f"🌤️ **Weather in {weather_data.get('name', city)}:**\n\nTemperature: {temp}°C\nConditions: {desc}\nHumidity: {humidity}%"
+                feels_like = weather_data["main"].get("feels_like", temp)
+                return f"🌤️ **Weather in {weather_data.get('name', city)}:**\n\n• Temperature: **{temp}°C** (Feels like: {feels_like}°C)\n• Conditions: **{desc}**\n• Humidity: **{humidity}%**"
             else:
                 return "❌ Could not fetch weather data. Please try again later."
         
@@ -349,7 +379,7 @@ class YAHBot:
                 price = quote.get("05. price", "N/A")
                 change = quote.get("09. change", "N/A")
                 change_percent = quote.get("10. change percent", "N/A")
-                return f"📈 **Apple Stock (AAPL):**\n\nPrice: ${price}\nChange: {change} ({change_percent})"
+                return f"📈 **Apple Stock (AAPL):**\n\n• Current Price: **${price}**\n• Change: **{change}** ({change_percent})"
             else:
                 return "❌ Could not fetch stock data. Please try again later."
         
@@ -359,7 +389,17 @@ class YAHBot:
                 btc_data = crypto_data["bitcoin"]
                 price = btc_data.get("usd", "N/A")
                 change = btc_data.get("usd_24h_change", "N/A")
-                return f"₿ **Bitcoin Price:**\n\nCurrent: ${price:,.2f}\n24h Change: {change:.2f}%"
+                if isinstance(price, (int, float)):
+                    price_str = f"${price:,.2f}"
+                else:
+                    price_str = str(price)
+                
+                if isinstance(change, (int, float)):
+                    change_str = f"{change:.2f}%"
+                else:
+                    change_str = str(change)
+                
+                return f"₿ **Bitcoin Price:**\n\n• Current: **{price_str}**\n• 24h Change: **{change_str}**"
             else:
                 return "❌ Could not fetch cryptocurrency data. Please try again later."
         
@@ -381,7 +421,7 @@ class YAHBot:
         if self.model and self.tokenizer:
             try:
                 # Prepare input with context
-                context = "You are YAH Bot, an AI assistant for YAH Tech. Be helpful, professional and focus on technology, entrepreneurship, and business development."
+                context = f"You are YAH Bot, an AI assistant for {COMPANY_INFO['name']}. You represent {PERSONAL_INFO['name']}, {PERSONAL_INFO['role']}. Be helpful, professional and focus on technology, entrepreneurship, venture studios, and business development. Our philosophy: {COMPANY_INFO['philosophy']}"
                 full_input = f"{context}\n\nConversation:\n" + "\n".join([f"Human: {msg['content']}\nAI: {msg['content']}" 
                                                                         for msg in conversation_history[-3:]]) + f"\nHuman: {user_input}\nAI:"
                 
@@ -390,7 +430,7 @@ class YAHBot:
                 with torch.no_grad():
                     outputs = self.model.generate(
                         inputs,
-                        max_length=150,
+                        max_length=200,
                         num_return_sequences=1,
                         temperature=0.7,
                         do_sample=True,
@@ -401,14 +441,14 @@ class YAHBot:
                 return response
                 
             except Exception as e:
-                return f"I encountered an error: {str(e)}. However, I can tell you that at YAH Tech, we're focused on building scalable systems that drive economic growth."
+                return f"I encountered an error: {str(e)}. However, I can tell you that at YAH Tech, we're focused on building scalable systems that drive economic growth through our venture studio model."
         
-        # Fallback responses
+      # Fallback responses
         fallback_responses = [
-            "That's an interesting question! At YAH Tech, we believe in learning, understanding, and creating solutions that matter.",
-            "I'd be happy to discuss that. As a venture studio, we focus on building technology-driven ventures that create both profit and societal value.",
-            "Great question! Our approach at YAH Tech is to develop futuristic solutions through careful evaluation and systematic creation.",
-            f"From my perspective as {PERSONAL_INFO['name']}, I focus on building systems that reshape the economic landscape through technology."
+            f"🌟 That's an interesting question! At {COMPANY_INFO['name']}, we believe in **{COMPANY_INFO['philosophy']}** to create solutions that matter.",
+            f"🚀 I'd be happy to discuss that. As a venture studio, we focus on building technology-driven ventures that create both **profit and societal value**.",
+            f"💡 Great question! Our approach at {COMPANY_INFO['name']} is to develop futuristic solutions through careful evaluation and systematic creation.",
+            f"🎯 From my perspective as {PERSONAL_INFO['name']}, I focus on building systems that reshape the economic landscape through innovative technology and scalable business models."
         ]
         return random.choice(fallback_responses)
 
@@ -420,89 +460,93 @@ def initialize_session_state():
         st.session_state.bot = YAHBot()
     if "conversation_started" not in st.session_state:
         st.session_state.conversation_started = False
-    if "api_status" not in st.session_state:
-        st.session_state.api_status = {}
 
 def display_api_status():
     """Display API status in sidebar"""
-    st.sidebar.subheader("API Status")
+    st.sidebar.subheader("🔌 API Status")
     
-    # Mock API status for demo
+    # Check API status based on environment variables
     statuses = {
-        "News API": "✅ Connected" if API_CONFIG["news_api"]["params"]["apiKey"] != "demo_key" else "⚠️ Demo Mode",
-        "Weather API": "✅ Connected" if API_CONFIG["weather_api"]["params"]["appid"] != "demo_key" else "⚠️ Demo Mode",
-        "Financial API": "✅ Connected" if API_CONFIG["financial_api"]["params"]["apikey"] != "demo_key" else "⚠️ Demo Mode",
-        "Crypto API": "✅ Connected",
-        "Internal API": "🔌 Not Connected"  # FastAPI server would be separate
+        "🤖 AI Model": "✅ Connected" if HUGGINGFACE_TOKEN else "⚠️ Using Standard Model",
+        "📰 News API": "✅ Connected" if os.environ.get("NEWS_API_KEY") and os.environ.get("NEWS_API_KEY") != "demo_key" else "🟡 Demo Mode",
+        "🌤️ Weather API": "✅ Connected" if os.environ.get("WEATHER_API_KEY") and os.environ.get("WEATHER_API_KEY") != "demo_key" else "🟡 Demo Mode",
+        "📈 Stock API": "✅ Connected" if os.environ.get("ALPHA_VANTAGE_KEY") and os.environ.get("ALPHA_VANTAGE_KEY") != "demo_key" else "🟡 Demo Mode",
+        "₿ Crypto API": "✅ Connected"
     }
     
     for api, status in statuses.items():
-        color = "🟢" if "✅" in status else "🟡" if "⚠️" in status else "🔴"
-        st.sidebar.write(f"{color} {api}: {status}")
+        st.sidebar.write(f"{api}: {status}")
 
 def main():
     # Page configuration
     st.set_page_config(
-        page_title="YAH Tech Chatbot",
-        page_icon="🤖",
-        layout="centered"
+        page_title="YAH Tech Assistant",
+        page_icon="🚀",
+        layout="centered",
+        initial_sidebar_state="expanded"
     )
     
     # Initialize session state
     initialize_session_state()
     
-    # Header
+    # Header with beautiful gradient
     col1, col2 = st.columns([1, 4])
     with col1:
-        st.image("https://via.placeholder.com/80/2196f3/ffffff?text=YT", width=80)
+        st.image("https://via.placeholder.com/100/2196f3/ffffff?text=YT", width=100)
     with col2:
-        st.title("YAH Tech Assistant")
-        st.caption("Powered by AI • Built for Innovation • API-Enabled")
+        st.markdown('<h1 class="header-gradient">YAH Tech Assistant</h1>', unsafe_allow_html=True)
+        st.caption("🚀 Powered by AI • 💼 Venture Studio • 🌍 Economic Innovation")
     
     st.markdown("---")
     
     # Sidebar
     with st.sidebar:
-        st.header("About YAH Tech")
+        st.header("🏢 About YAH Tech")
         st.markdown(f"""
-        **Founder**: {PERSONAL_INFO['name']}  
-        **Role**: {PERSONAL_INFO['role']}  
+        **👤 Founder**: {PERSONAL_INFO['name']}  
+        **💼 Role**: {PERSONAL_INFO['role']}  
         
-        **Company**: {COMPANY_INFO['name']}  
-        **Focus**: {COMPANY_INFO['type']}  
-        **Stage**: {COMPANY_INFO['stage']}
+        **🏢 Company**: {COMPANY_INFO['name']}  
+        **🎯 Focus**: {COMPANY_INFO['type']}  
+        **📈 Stage**: {COMPANY_INFO['stage']}
         
-        **Philosophy**: *{COMPANY_INFO['philosophy']}*
+        **💡 Philosophy**: *{COMPANY_INFO['philosophy']}*
         """)
         
         # API Status
         display_api_status()
         
         st.markdown("---")
-        st.subheader("API Capabilities")
+        st.subheader("🔧 Quick Actions")
         st.markdown("""
-        Ask me to:
-        - **Get tech news** 📰
-        - **Check weather** 🌤️
-        - **Stock prices** 📈
-        - **Crypto data** ₿
-        - **YAH Tech info** 🏢
-        - **Business advice** 💼
+        Try asking me:
+        - *"What's the latest tech news?"* 📰
+        - *"Weather in Lagos"* 🌤️  
+        - *"Apple stock price"* 📈
+        - *"Bitcoin price"* ₿
+        - *"Tell me about YAH Tech"* 🏢
+        - *"What are your goals?"* 🎯
         """)
         
-        st.markdown("---")
-        st.subheader("Quick Actions")
-        if st.button("🔄 Test APIs"):
-            st.info("API testing feature would be implemented here")
-        
-        if st.button("🗑️ Clear Chat"):
+        if st.button("🗑️ Clear Chat History", use_container_width=True):
             st.session_state.messages = []
             st.session_state.conversation_started = False
             st.rerun()
     
     # Chat interface
     if not st.session_state.conversation_started:
-        st.info(f"👋 Hello! I'm YAH Bot, your AI assistant for everything related to {PERSONAL_INFO['name']} and {COMPANY_INFO['name']}. I can also fetch real-time data through APIs! How can I help you today?")
+        welcome_msg = f"""
+        👋 **Hello! I'm YAH Bot, your AI assistant for everything related to {PERSONAL_INFO['name']} and {COMPANY_INFO['name']}!** 
+        
+        I can help you with:
+        • 📊 Company information and venture studio details
+        • 🎯 Adedoyin's vision and goals  
+        • 🌐 Real-time data (news, weather, stocks, crypto)
+        • 💡 Business and technology insights
+        
+        **How can I assist you today?** 🚀
+        """
+        st.info(welcome_msg)
         st.session_state.conversation_started = True
     
     # Display chat messages
@@ -511,7 +555,7 @@ def main():
             st.markdown(message["content"])
     
     # Chat input
-    if prompt := st.chat_input("Ask me about YAH Tech, get news, check weather, or anything else..."):
+    if prompt := st.chat_input("💬 Ask me about YAH Tech, get real-time data, or anything else..."):
         # Add user message to chat history
         st.session_state.messages.append({"role": "user", "content": prompt})
         with st.chat_message("user"):
@@ -519,10 +563,15 @@ def main():
         
         # Generate and display assistant response
         with st.chat_message("assistant"):
-            with st.spinner("🤔 Thinking..."):
+            with st.spinner("🤔 Processing your request..."):
                 # Use asyncio to handle async API calls
-                response = asyncio.run(st.session_state.bot.generate_response(prompt, st.session_state.messages))
-                st.markdown(response)
+                try:
+                    response = asyncio.run(st.session_state.bot.generate_response(prompt, st.session_state.messages))
+                    st.markdown(response)
+                except Exception as e:
+                    error_msg = f"⚠️ I encountered an issue: {str(e)}\n\nBut at YAH Tech, we always find solutions! Feel free to ask me something else about our venture studio or business philosophy."
+                    st.markdown(error_msg)
+                    response = error_msg
         
         st.session_state.messages.append({"role": "assistant", "content": response})