fix: etherscan_tool error fix

.env.example

@@ -1,8 +1,14 @@
-# Google Gemini API Key (Required)
-GEMINI_API_KEY=AIzaSyCgOdLBoDEeG7vONzCZgm1l9dJFyIYrZOw
+# Required for full AI functionality
+GEMINI_API_KEY=your_google_gemini_api_key_here
 
-# CoinGecko API Key (Optional - for higher rate limits)
+# Optional - Enhanced data sources  
 COINGECKO_API_KEY=your_coingecko_api_key_here
+ETHERSCAN_API_KEY=your_etherscan_api_key_here
+DEFILLAMA_API_KEY=your_defillama_api_key_here
 
-# CryptoCompare API Key (Optional - for additional data sources)
-CRYPTOCOMPARE_API_KEY=your_cryptocompare_api_key_here
+# Optional - Logging configuration
+LOG_LEVEL=INFO
+
+# Optional - AIRAA Integration
+AIRAA_ENDPOINT=your_airaa_endpoint_here
+AIRAA_API_KEY=your_airaa_api_key_here

Dockerfile

@@ -1,18 +1,37 @@
+# Use Python 3.11 slim image for HuggingFace Spaces
 FROM python:3.11-slim
 
+# Set working directory
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
 
+# Copy requirements first for better caching
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
 
+# Install Python dependencies
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
 COPY . .
 
-EXPOSE 7860
+# Create necessary directories
+RUN mkdir -p logs cache
 
+# Set environment variables for HuggingFace Spaces
 ENV PYTHONPATH=/app
-ENV GRADIO_SERVER_NAME=0.0.0.0
-ENV GRADIO_SERVER_PORT=7860
+ENV PYTHONUNBUFFERED=1
+
+# Expose port 7860 (HuggingFace Spaces default)
+EXPOSE 7860
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
 
-CMD ["python", "app.py"]
+# Run the application
+CMD ["python", "app_fastapi.py"]

app_fastapi.py → app.py

@@ -9,6 +9,8 @@ from datetime import datetime
 from typing import List, Dict, Any, Optional
 import os
 from dotenv import load_dotenv
+import plotly
+import plotly.graph_objects as go
 
 load_dotenv()
 
@@ -16,16 +18,17 @@ from src.agent.research_agent import Web3ResearchAgent
 from src.api.airaa_integration import AIRAAIntegration  
 from src.utils.logger import get_logger
 from src.utils.config import config
+from src.visualizations import CryptoVisualizations
 
 logger = get_logger(__name__)
 
 app = FastAPI(
     title="Web3 Research Co-Pilot",
-    description="AI-powered cryptocurrency research assistant",
-    version="1.0.0"
+    description="Professional cryptocurrency research assistant",
+    version="2.0.0"
 )
 
-# Pydantic models for request/response
+# Pydantic models
 class QueryRequest(BaseModel):
     query: str
     chat_history: Optional[List[Dict[str, str]]] = []
@@ -35,93 +38,140 @@ class QueryResponse(BaseModel):
     response: str
     sources: Optional[List[str]] = []
     metadata: Optional[Dict[str, Any]] = {}
+    visualizations: Optional[List[str]] = []
     error: Optional[str] = None
 
 class Web3CoPilotService:
     def __init__(self):
         try:
-            logger.info("🚀 Initializing Web3CoPilotService...")
-            logger.info(f"📋 GEMINI_API_KEY configured: {'Yes' if config.GEMINI_API_KEY else 'No'}")
+            logger.info("Initializing Web3 Research Co-Pilot...")
             
             if config.GEMINI_API_KEY:
-                logger.info("🤖 Initializing AI agent...")
+                logger.info("Initializing AI research agent...")
                 self.agent = Web3ResearchAgent()
-                logger.info("✅ AI agent initialized successfully")
+                logger.info("AI research agent initialized")
             else:
-                logger.warning("⚠️ GEMINI_API_KEY not found - AI features disabled")
+                logger.warning("GEMINI_API_KEY not configured - limited functionality")
                 self.agent = None
             
-            logger.info("🔗 Initializing AIRAA integration...")
+            logger.info("Initializing integrations...")
             self.airaa = AIRAAIntegration()
-            logger.info(f"🔗 AIRAA integration: {'Enabled' if self.airaa.enabled else 'Disabled'}")
             
             self.enabled = bool(config.GEMINI_API_KEY)
-            logger.info(f"🎯 Web3CoPilotService initialized successfully (AI enabled: {self.enabled})")
+            self.visualizer = CryptoVisualizations()
+            
+            logger.info(f"Service initialized (AI enabled: {self.enabled})")
             
         except Exception as e:
-            logger.error(f"❌ Service initialization failed: {e}")
+            logger.error(f"Service initialization failed: {e}")
             self.agent = None
             self.airaa = None
             self.enabled = False
+            self.visualizer = CryptoVisualizations()
     
     async def process_query(self, query: str) -> QueryResponse:
-        logger.info(f"🔍 Processing query: {query[:50]}{'...' if len(query) > 50 else ''}")
+        """Process research query with visualizations"""
+        logger.info(f"Processing query: {query[:100]}...")
         
         if not query.strip():
-            logger.warning("⚠️ Empty query received")
-            return QueryResponse(success=False, response="Please enter a query.", error="Empty query")
+            return QueryResponse(
+                success=False, 
+                response="Please provide a research query.", 
+                error="Empty query"
+            )
         
         try:
             if not self.enabled:
-                logger.info("🔧 AI disabled - providing limited response")
-                response = """⚠️ **AI Agent Disabled**: GEMINI_API_KEY not configured.
+                response = """**Research Assistant - Limited Mode**
 
-**Limited Data Available:**
-- CoinGecko API (basic crypto data)
-- DeFiLlama API (DeFi protocols) 
-- Etherscan API (gas prices)
+API access available for basic cryptocurrency data:
+• Market prices and statistics
+• DeFi protocol information  
+• Network gas fees
 
-Please configure GEMINI_API_KEY for full AI analysis."""
-                return QueryResponse(success=True, response=response, sources=["Configuration"])
+Configure GEMINI_API_KEY environment variable for full AI analysis."""
+                return QueryResponse(success=True, response=response, sources=["System"])
             
-            logger.info("🤖 Sending query to AI agent...")
+            logger.info("Processing with AI research agent...")
             result = await self.agent.research_query(query)
-            logger.info(f"✅ AI agent responded: {result.get('success', False)}")
             
             if result.get("success"):
-                response = result.get("result", "No response generated")
+                response = result.get("result", "No analysis generated")
                 sources = result.get("sources", [])
                 metadata = result.get("metadata", {})
                 
+                # Generate visualizations if relevant data is available
+                visualizations = []
+                if metadata:
+                    vis_html = await self._generate_visualizations(metadata, query)
+                    if vis_html:
+                        visualizations.append(vis_html)
+                
                 # Send to AIRAA if enabled
                 if self.airaa and self.airaa.enabled:
                     try:
-                        logger.info("🔗 Sending data to AIRAA...")
                         await self.airaa.send_research_data(query, response)
-                        logger.info("✅ Data sent to AIRAA successfully")
+                        logger.info("Data sent to AIRAA")
                     except Exception as e:
-                        logger.warning(f"⚠️ AIRAA integration failed: {e}")
+                        logger.warning(f"AIRAA integration failed: {e}")
                 
-                logger.info("✅ Query processed successfully")
-                return QueryResponse(success=True, response=response, sources=sources, metadata=metadata)
+                return QueryResponse(
+                    success=True, 
+                    response=response, 
+                    sources=sources, 
+                    metadata=metadata,
+                    visualizations=visualizations
+                )
             else:
-                error_msg = result.get("error", "Research failed. Please try again.")
-                logger.error(f"❌ AI agent failed: {error_msg}")
+                error_msg = result.get("error", "Research analysis failed")
+                logger.error(f"Research failed: {error_msg}")
                 return QueryResponse(success=False, response=error_msg, error=error_msg)
             
         except Exception as e:
-            logger.error(f"❌ Query processing error: {e}")
-            error_msg = f"Error processing query: {str(e)}"
+            logger.error(f"Query processing error: {e}")
+            error_msg = f"Processing error: {str(e)}"
             return QueryResponse(success=False, response=error_msg, error=error_msg)
+    
+    async def _generate_visualizations(self, metadata: Dict[str, Any], query: str) -> Optional[str]:
+        """Generate visualizations based on query and metadata"""
+        try:
+            # Check for price data
+            if 'price_data' in metadata:
+                symbol = self._extract_symbol_from_query(query)
+                fig = self.visualizer.create_price_chart(metadata['price_data'], symbol)
+                return plotly.io.to_html(fig, include_plotlyjs='cdn', div_id='price_chart')
+            
+            # Check for market data
+            elif 'market_data' in metadata:
+                fig = self.visualizer.create_market_overview(metadata['market_data'])
+                return plotly.io.to_html(fig, include_plotlyjs='cdn', div_id='market_overview')
+            
+            # Check for DeFi data
+            elif 'defi_data' in metadata:
+                fig = self.visualizer.create_defi_tvl_chart(metadata['defi_data'])
+                return plotly.io.to_html(fig, include_plotlyjs='cdn', div_id='defi_chart')
+            
+            return None
+            
+        except Exception as e:
+            logger.error(f"Visualization generation failed: {e}")
+            return None
+    
+    def _extract_symbol_from_query(self, query: str) -> str:
+        """Extract cryptocurrency symbol from query"""
+        symbols = ['BTC', 'ETH', 'ADA', 'SOL', 'AVAX', 'MATIC', 'DOT', 'LINK']
+        query_upper = query.upper()
+        for symbol in symbols:
+            if symbol in query_upper:
+                return symbol
+        return 'BTC'  # Default
 
 # Initialize service
-logger.info("🚀 Starting Web3 Research Co-Pilot...")
 service = Web3CoPilotService()
 
-# API Routes
 @app.get("/", response_class=HTMLResponse)
 async def get_homepage(request: Request):
-    logger.info("📄 Serving homepage")
+    """Serve minimalist, professional interface"""
     html_content = """
     <!DOCTYPE html>
     <html lang="en">
@@ -129,434 +179,593 @@ async def get_homepage(request: Request):
         <meta charset="UTF-8">
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
         <title>Web3 Research Co-Pilot</title>
-        <link rel="icon" href="data:image/svg+xml,<svg xmlns=%22http://www.w3.org/2000/svg%22 viewBox=%220 0 100 100%22><text y=%22.9em%22 font-size=%2290%22>🚀</text></svg>">
+        <link rel="icon" href="data:image/svg+xml,<svg xmlns=%22http://www.w3.org/2000/svg%22 viewBox=%220 0 24 24%22><path fill=%22%2300d4aa%22 d=%22M12 2L2 7v10c0 5.5 3.8 7.7 9 9 5.2-1.3 9-3.5 9-9V7l-10-5z%22/></svg>">
+        
         <style>
-            * { margin: 0; padding: 0; box-sizing: border-box; }
-            body { 
-                font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Oxygen', 'Ubuntu', 'Cantarell', sans-serif; 
-                background: linear-gradient(135deg, #0f1419 0%, #1a1f2e 100%); 
-                color: #e6e6e6; 
+            :root {
+                --primary: #0066ff;
+                --primary-dark: #0052cc;
+                --accent: #00d4aa;
+                --background: #000000;
+                --surface: #111111;
+                --surface-elevated: #1a1a1a;
+                --text: #ffffff;
+                --text-secondary: #a0a0a0;
+                --text-muted: #666666;
+                --border: rgba(255, 255, 255, 0.08);
+                --border-focus: rgba(0, 102, 255, 0.3);
+                --shadow: rgba(0, 0, 0, 0.4);
+                --success: #00d4aa;
+                --warning: #ffa726;
+                --error: #f44336;
+            }
+
+            * {
+                margin: 0;
+                padding: 0;
+                box-sizing: border-box;
+            }
+
+            body {
+                font-family: -apple-system, BlinkMacSystemFont, 'SF Pro Display', system-ui, sans-serif;
+                background: var(--background);
+                color: var(--text);
+                line-height: 1.5;
                 min-height: 100vh;
-                overflow-x: hidden;
-            }
-            .container { max-width: 1200px; margin: 0 auto; padding: 20px; }
-            .header { 
-                text-align: center; 
-                margin-bottom: 30px; 
-                background: linear-gradient(135deg, #00d4aa, #4a9eff);
-                -webkit-background-clip: text;
-                -webkit-text-fill-color: transparent;
-                background-clip: text;
-            }
-            .header h1 { 
-                font-size: 3em; 
-                margin-bottom: 10px; 
-                font-weight: 700;
-                text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
-            }
-            .header p { 
-                color: #b0b0b0; 
-                font-size: 1.2em; 
-                font-weight: 300;
-            }
-            .status { 
-                padding: 15px; 
-                border-radius: 12px; 
-                margin-bottom: 25px; 
-                text-align: center; 
-                font-weight: 500;
-                box-shadow: 0 4px 15px rgba(0,0,0,0.2);
-                transition: all 0.3s ease;
+                font-weight: 400;
+                -webkit-font-smoothing: antialiased;
+                -moz-osx-font-smoothing: grayscale;
+            }
+
+            .container {
+                max-width: 1000px;
+                margin: 0 auto;
+                padding: 2rem 1.5rem;
+            }
+
+            .header {
+                text-align: center;
+                margin-bottom: 2.5rem;
             }
-            .status.enabled { 
-                background: linear-gradient(135deg, #1a4d3a, #2a5d4a); 
-                border: 2px solid #00d4aa; 
-                color: #00d4aa; 
+
+            .header h1 {
+                font-size: 2.25rem;
+                font-weight: 600;
+                color: var(--text);
+                margin-bottom: 0.5rem;
+                letter-spacing: -0.025em;
             }
-            .status.disabled { 
-                background: linear-gradient(135deg, #4d1a1a, #5d2a2a); 
-                border: 2px solid #ff6b6b; 
-                color: #ff6b6b; 
+
+            .header .brand {
+                color: var(--primary);
             }
+
+            .header p {
+                color: var(--text-secondary);
+                font-size: 1rem;
+                font-weight: 400;
+            }
+
+            .status {
+                background: var(--surface);
+                border: 1px solid var(--border);
+                border-radius: 12px;
+                padding: 1rem 1.5rem;
+                margin-bottom: 2rem;
+                text-align: center;
+                transition: all 0.2s ease;
+            }
+
+            .status.online {
+                border-color: var(--success);
+                background: linear-gradient(135deg, rgba(0, 212, 170, 0.05), rgba(0, 212, 170, 0.02));
+            }
+
+            .status.offline {
+                border-color: var(--error);
+                background: linear-gradient(135deg, rgba(244, 67, 54, 0.05), rgba(244, 67, 54, 0.02));
+            }
+
             .status.checking {
-                background: linear-gradient(135deg, #3a3a1a, #4a4a2a); 
-                border: 2px solid #ffdd59; 
-                color: #ffdd59;
-                animation: pulse 1.5s infinite;
+                border-color: var(--warning);
+                background: linear-gradient(135deg, rgba(255, 167, 38, 0.05), rgba(255, 167, 38, 0.02));
+                animation: pulse 2s infinite;
             }
+
             @keyframes pulse {
-                0% { opacity: 1; }
-                50% { opacity: 0.7; }
-                100% { opacity: 1; }
-            }
-            .chat-container { 
-                background: rgba(26, 26, 26, 0.8); 
-                border-radius: 16px; 
-                padding: 25px; 
-                margin-bottom: 25px; 
-                backdrop-filter: blur(10px);
-                border: 1px solid rgba(255,255,255,0.1);
-                box-shadow: 0 8px 32px rgba(0,0,0,0.3);
-            }
-            .chat-messages { 
-                height: 450px; 
-                overflow-y: auto; 
-                background: rgba(10, 10, 10, 0.6); 
-                border-radius: 12px; 
-                padding: 20px; 
-                margin-bottom: 20px;
-                border: 1px solid rgba(255,255,255,0.05);
-            }
-            .chat-messages::-webkit-scrollbar { width: 6px; }
-            .chat-messages::-webkit-scrollbar-track { background: #2a2a2a; border-radius: 3px; }
-            .chat-messages::-webkit-scrollbar-thumb { background: #555; border-radius: 3px; }
-            .chat-messages::-webkit-scrollbar-thumb:hover { background: #777; }
-            .message { 
-                margin-bottom: 20px; 
-                padding: 16px; 
-                border-radius: 12px; 
-                transition: all 0.3s ease;
+                0%, 100% { opacity: 1; }
+                50% { opacity: 0.8; }
+            }
+
+            .chat-interface {
+                background: var(--surface);
+                border: 1px solid var(--border);
+                border-radius: 16px;
+                overflow: hidden;
+                margin-bottom: 2rem;
+                backdrop-filter: blur(20px);
+            }
+
+            .chat-messages {
+                height: 480px;
+                overflow-y: auto;
+                padding: 2rem;
+                background: linear-gradient(180deg, var(--background), var(--surface));
+            }
+
+            .chat-messages::-webkit-scrollbar {
+                width: 3px;
+            }
+
+            .chat-messages::-webkit-scrollbar-track {
+                background: transparent;
+            }
+
+            .chat-messages::-webkit-scrollbar-thumb {
+                background: var(--border);
+                border-radius: 2px;
+            }
+
+            .message {
+                margin-bottom: 2rem;
+                opacity: 0;
+                animation: messageSlide 0.4s cubic-bezier(0.2, 0, 0.2, 1) forwards;
+            }
+
+            @keyframes messageSlide {
+                from { 
+                    opacity: 0; 
+                    transform: translateY(20px) scale(0.98); 
+                }
+                to { 
+                    opacity: 1; 
+                    transform: translateY(0) scale(1); 
+                }
+            }
+
+            .message.user {
+                text-align: right;
+            }
+
+            .message.assistant {
+                text-align: left;
+            }
+
+            .message-content {
+                display: inline-block;
+                max-width: 75%;
+                padding: 1.25rem 1.5rem;
+                border-radius: 24px;
+                font-size: 0.95rem;
+                line-height: 1.6;
                 position: relative;
             }
-            .message:hover { transform: translateY(-1px); box-shadow: 0 4px 12px rgba(0,0,0,0.2); }
-            .message.user { 
-                background: linear-gradient(135deg, #2a2a3a, #3a3a4a); 
-                border-left: 4px solid #00d4aa; 
-                margin-left: 50px;
-            }
-            .message.assistant { 
-                background: linear-gradient(135deg, #1a2a1a, #2a3a2a); 
-                border-left: 4px solid #4a9eff; 
-                margin-right: 50px;
-            }
-            .message .sender { 
-                font-weight: 600; 
-                margin-bottom: 8px; 
-                font-size: 0.9em; 
+
+            .message.user .message-content {
+                background: linear-gradient(135deg, var(--primary), var(--primary-dark));
+                color: #ffffff;
+                border-bottom-right-radius: 8px;
+                box-shadow: 0 4px 12px rgba(0, 102, 255, 0.2);
+            }
+
+            .message.assistant .message-content {
+                background: var(--surface-elevated);
+                color: var(--text);
+                border-bottom-left-radius: 8px;
+                border: 1px solid var(--border);
+            }
+
+            .message-meta {
+                font-size: 0.75rem;
+                color: var(--text-muted);
+                margin-top: 0.5rem;
+                font-weight: 500;
+            }
+
+            .sources {
+                margin-top: 1rem;
+                padding-top: 1rem;
+                border-top: 1px solid var(--border);
+                font-size: 0.8rem;
+                color: var(--text-secondary);
+            }
+
+            .sources span {
+                display: inline-block;
+                background: rgba(0, 102, 255, 0.1);
+                border: 1px solid rgba(0, 102, 255, 0.2);
+                padding: 0.25rem 0.75rem;
+                border-radius: 6px;
+                margin: 0.25rem 0.5rem 0.25rem 0;
+                font-weight: 500;
+                font-size: 0.75rem;
+            }
+
+            .input-area {
+                padding: 2rem;
+                background: linear-gradient(180deg, var(--surface), var(--surface-elevated));
+                border-top: 1px solid var(--border);
+            }
+
+            .input-container {
                 display: flex;
-                align-items: center;
-                gap: 8px;
-            }
-            .message.user .sender { color: #00d4aa; }
-            .message.assistant .sender { color: #4a9eff; }
-            .message .content { line-height: 1.6; }
-            .input-container { 
-                display: flex; 
-                gap: 12px; 
+                gap: 1rem;
                 align-items: stretch;
             }
-            .input-container input { 
-                flex: 1; 
-                padding: 16px; 
-                border: 2px solid #333; 
-                background: rgba(42, 42, 42, 0.8); 
-                color: #e6e6e6; 
-                border-radius: 12px; 
-                font-size: 16px;
-                backdrop-filter: blur(10px);
-                transition: all 0.3s ease;
-            }
-            .input-container input:focus { 
-                outline: none; 
-                border-color: #00d4aa; 
-                box-shadow: 0 0 0 3px rgba(0, 212, 170, 0.2);
-            }
-            .input-container input::placeholder { color: #888; }
-            .input-container button { 
-                padding: 16px 24px; 
-                background: linear-gradient(135deg, #00d4aa, #00b894); 
-                color: #000; 
-                border: none; 
-                border-radius: 12px; 
-                cursor: pointer; 
+
+            .input-field {
+                flex: 1;
+                padding: 1rem 1.5rem;
+                background: var(--background);
+                border: 2px solid var(--border);
+                border-radius: 28px;
+                color: var(--text);
+                font-size: 0.95rem;
+                outline: none;
+                transition: all 0.2s cubic-bezier(0.2, 0, 0.2, 1);
+                font-weight: 400;
+            }
+
+            .input-field:focus {
+                border-color: var(--primary);
+                box-shadow: 0 0 0 4px var(--border-focus);
+                background: var(--surface);
+            }
+
+            .input-field::placeholder {
+                color: var(--text-muted);
+                font-weight: 400;
+            }
+
+            .send-button {
+                padding: 1rem 2rem;
+                background: linear-gradient(135deg, var(--primary), var(--primary-dark));
+                color: #ffffff;
+                border: none;
+                border-radius: 28px;
                 font-weight: 600;
-                font-size: 16px;
-                transition: all 0.3s ease;
-                white-space: nowrap;
+                cursor: pointer;
+                transition: all 0.2s cubic-bezier(0.2, 0, 0.2, 1);
+                font-size: 0.95rem;
+                box-shadow: 0 4px 12px rgba(0, 102, 255, 0.2);
             }
-            .input-container button:hover:not(:disabled) { 
-                background: linear-gradient(135deg, #00b894, #00a085); 
+
+            .send-button:hover:not(:disabled) {
                 transform: translateY(-2px);
-                box-shadow: 0 4px 12px rgba(0, 212, 170, 0.3);
+                box-shadow: 0 8px 24px rgba(0, 102, 255, 0.3);
             }
-            .input-container button:active { transform: translateY(0); }
-            .input-container button:disabled { 
-                background: #666; 
-                cursor: not-allowed; 
+
+            .send-button:active {
+                transform: translateY(0);
+            }
+
+            .send-button:disabled {
+                opacity: 0.6;
+                cursor: not-allowed;
                 transform: none;
-                box-shadow: none;
-            }
-            .examples { 
-                display: grid; 
-                grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); 
-                gap: 15px; 
-                margin-top: 25px; 
-            }
-            .example-btn { 
-                padding: 16px; 
-                background: linear-gradient(135deg, #2a2a3a, #3a3a4a); 
-                border: 2px solid #444; 
-                border-radius: 12px; 
-                cursor: pointer; 
-                text-align: center; 
-                transition: all 0.3s ease;
-                font-weight: 500;
+                box-shadow: 0 4px 12px rgba(0, 102, 255, 0.1);
+            }
+
+            .examples {
+                display: grid;
+                grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
+                gap: 1rem;
+                margin-top: 1rem;
+            }
+
+            .example {
+                background: linear-gradient(135deg, var(--surface), var(--surface-elevated));
+                border: 1px solid var(--border);
+                border-radius: 12px;
+                padding: 1.5rem;
+                cursor: pointer;
+                transition: all 0.3s cubic-bezier(0.2, 0, 0.2, 1);
                 position: relative;
                 overflow: hidden;
             }
-            .example-btn:before {
+
+            .example::before {
                 content: '';
                 position: absolute;
                 top: 0;
                 left: -100%;
                 width: 100%;
                 height: 100%;
-                background: linear-gradient(90deg, transparent, rgba(0, 212, 170, 0.1), transparent);
-                transition: left 0.5s;
-            }
-            .example-btn:hover:before { left: 100%; }
-            .example-btn:hover { 
-                background: linear-gradient(135deg, #3a3a4a, #4a4a5a); 
-                border-color: #00d4aa; 
-                transform: translateY(-3px);
-                box-shadow: 0 6px 20px rgba(0, 212, 170, 0.2);
-            }
-            .loading { 
-                color: #ffdd59; 
-                font-style: italic; 
-                display: flex;
+                background: linear-gradient(90deg, transparent, rgba(0, 102, 255, 0.05), transparent);
+                transition: left 0.5s ease;
+            }
+
+            .example:hover::before {
+                left: 100%;
+            }
+
+            .example:hover {
+                border-color: var(--primary);
+                transform: translateY(-4px);
+                box-shadow: 0 12px 32px rgba(0, 0, 0, 0.2);
+                background: linear-gradient(135deg, var(--surface-elevated), var(--surface));
+            }
+
+            .example-title {
+                font-weight: 600;
+                color: var(--text);
+                margin-bottom: 0.5rem;
+                font-size: 0.95rem;
+            }
+
+            .example-desc {
+                font-size: 0.85rem;
+                color: var(--text-secondary);
+                font-weight: 400;
+            }
+
+            .loading {
+                display: inline-flex;
                 align-items: center;
-                gap: 8px;
+                gap: 0.5rem;
+                color: var(--text-secondary);
+                font-weight: 500;
             }
-            .loading:after {
+
+            .loading::after {
                 content: '';
-                width: 12px;
-                height: 12px;
-                border: 2px solid #ffdd59;
-                border-top: 2px solid transparent;
+                width: 14px;
+                height: 14px;
+                border: 2px solid currentColor;
+                border-top-color: transparent;
                 border-radius: 50%;
                 animation: spin 1s linear infinite;
             }
+
             @keyframes spin {
-                0% { transform: rotate(0deg); }
-                100% { transform: rotate(360deg); }
+                to { transform: rotate(360deg); }
             }
-            .sources { 
-                margin-top: 12px; 
-                font-size: 0.85em; 
-                color: #999; 
-                display: flex;
-                flex-wrap: wrap;
-                gap: 6px;
-            }
-            .sources .label { margin-right: 8px; font-weight: 600; }
-            .sources span { 
-                background: rgba(51, 51, 51, 0.8); 
-                padding: 4px 8px; 
-                border-radius: 6px; 
-                font-size: 0.8em;
-                border: 1px solid #555;
-            }
-            .welcome-message {
-                background: linear-gradient(135deg, #1a2a4a, #2a3a5a);
-                border-left: 4px solid #4a9eff;
+
+            .visualization-container {
+                margin: 1.5rem 0;
+                background: var(--surface-elevated);
                 border-radius: 12px;
-                padding: 16px;
-                margin-bottom: 20px;
-                text-align: center;
+                padding: 1.5rem;
+                border: 1px solid var(--border);
             }
-            .footer {
+
+            .welcome {
                 text-align: center;
-                margin-top: 30px;
-                color: #666;
-                font-size: 0.9em;
+                padding: 4rem 2rem;
+                color: var(--text-secondary);
+            }
+
+            .welcome h3 {
+                font-size: 1.25rem;
+                font-weight: 600;
+                margin-bottom: 0.5rem;
+                color: var(--text);
+            }
+
+            .welcome p {
+                font-size: 0.95rem;
+                font-weight: 400;
+            }
+
+            @media (max-width: 768px) {
+                .container {
+                    padding: 1rem;
+                }
+                
+                .header h1 {
+                    font-size: 1.75rem;
+                }
+                
+                .chat-messages {
+                    height: 400px;
+                    padding: 1.5rem;
+                }
+                
+                .message-content {
+                    max-width: 85%;
+                    padding: 1rem 1.25rem;
+                }
+                
+                .input-area {
+                    padding: 1.5rem;
+                }
+                
+                .input-container {
+                    flex-direction: column;
+                    gap: 0.75rem;
+                }
+                
+                .send-button {
+                    align-self: stretch;
+                }
+                
+                .examples {
+                    grid-template-columns: 1fr;
+                }
             }
         </style>
     </head>
     <body>
         <div class="container">
             <div class="header">
-                <h1>🚀 Web3 Research Co-Pilot</h1>
-                <p>AI-powered cryptocurrency research assistant</p>
+                <h1><span class="brand">Web3</span> Research Co-Pilot</h1>
+                <p>Professional cryptocurrency analysis and market intelligence</p>
             </div>
-            
+
             <div id="status" class="status checking">
-                <span>🔄 Checking system status...</span>
+                <span>Initializing research systems...</span>
             </div>
-            
-            <div class="chat-container">
+
+            <div class="chat-interface">
                 <div id="chatMessages" class="chat-messages">
-                    <div class="welcome-message">
-                        <div class="sender">🤖 AI Research Assistant</div>
-                        <div>👋 Welcome! I'm your Web3 Research Co-Pilot. Ask me anything about cryptocurrency markets, DeFi protocols, blockchain analysis, or trading insights.</div>
+                    <div class="welcome">
+                        <h3>Welcome to Web3 Research Co-Pilot</h3>
+                        <p>Ask about market trends, DeFi protocols, or blockchain analytics</p>
                     </div>
                 </div>
-                <div class="input-container">
-                    <input type="text" id="queryInput" placeholder="Ask about Bitcoin, Ethereum, DeFi yields, market analysis..." maxlength="500">
-                    <button id="sendBtn" onclick="sendQuery()">🚀 Research</button>
+                <div class="input-area">
+                    <div class="input-container">
+                        <input 
+                            type="text" 
+                            id="queryInput" 
+                            class="input-field"
+                            placeholder="Research Bitcoin trends, analyze DeFi yields, compare protocols..."
+                            maxlength="500"
+                        >
+                        <button id="sendBtn" class="send-button">Research</button>
+                    </div>
                 </div>
             </div>
-            
+
             <div class="examples">
-                <div class="example-btn" onclick="setQuery('What is the current Bitcoin price and market sentiment?')">
-                    📈 Bitcoin Analysis
+                <div class="example" onclick="setQuery('Analyze Bitcoin price trends and institutional adoption patterns')">
+                    <div class="example-title">Market Analysis</div>
+                    <div class="example-desc">Bitcoin trends, institutional flows, and market sentiment</div>
                 </div>
-                <div class="example-btn" onclick="setQuery('Show me the top DeFi protocols by TVL')">
-                    🏦 DeFi Overview
+                <div class="example" onclick="setQuery('Compare top DeFi protocols by TVL, yield, and risk metrics')">
+                    <div class="example-title">DeFi Intelligence</div>
+                    <div class="example-desc">Protocol comparison, yield analysis, and risk assessment</div>
                 </div>
-                <div class="example-btn" onclick="setQuery('What are the trending cryptocurrencies today?')">
-                    🔥 Trending Coins
+                <div class="example" onclick="setQuery('Evaluate Ethereum Layer 2 scaling solutions and adoption metrics')">
+                    <div class="example-title">Layer 2 Research</div>
+                    <div class="example-desc">Scaling solutions, transaction costs, and ecosystem growth</div>
                 </div>
-                <div class="example-btn" onclick="setQuery('Analyze Ethereum gas prices and network activity')">
-                    ⛽ Gas Tracker
+                <div class="example" onclick="setQuery('Identify optimal yield farming strategies across multiple chains')">
+                    <div class="example-title">Yield Optimization</div>
+                    <div class="example-desc">Cross-chain opportunities, APY tracking, and risk analysis</div>
                 </div>
-                <div class="example-btn" onclick="setQuery('Find the best yield farming opportunities')">
-                    🌾 Yield Farming
-                </div>
-                <div class="example-btn" onclick="setQuery('Compare Solana vs Ethereum ecosystems')">
-                    ⚖️ Ecosystem Compare
-                </div>
-            </div>
-            
-            <div class="footer">
-                <p>Powered by AI • Real-time Web3 data • Built with ❤️</p>
             </div>
         </div>
-        
+
         <script>
             let chatHistory = [];
-            
+            let messageCount = 0;
+
             async function checkStatus() {
                 try {
-                    console.log('🔍 Checking system status...');
                     const response = await fetch('/status');
                     const status = await response.json();
-                    console.log('📊 Status received:', status);
                     
                     const statusDiv = document.getElementById('status');
                     
                     if (status.enabled && status.gemini_configured) {
-                        statusDiv.className = 'status enabled';
+                        statusDiv.className = 'status online';
                         statusDiv.innerHTML = `
-                            <span>✅ AI Research Agent: Online</span><br>
-                            <small>Tools available: ${status.tools_available.join(', ')}</small>
+                            <span>Research systems online</span>
+                            <div style="margin-top: 0.5rem; font-size: 0.85rem; opacity: 0.8;">
+                                Tools: ${status.tools_available.join(' • ')}
+                            </div>
                         `;
-                        console.log('✅ System fully operational');
                     } else {
-                        statusDiv.className = 'status disabled';
+                        statusDiv.className = 'status offline';
                         statusDiv.innerHTML = `
-                            <span>⚠️ Limited Mode: GEMINI_API_KEY not configured</span><br>
-                            <small>Basic data available: ${status.tools_available.join(', ')}</small>
+                            <span>Limited mode - Configure GEMINI_API_KEY for full functionality</span>
+                            <div style="margin-top: 0.5rem; font-size: 0.85rem; opacity: 0.8;">
+                                Available: ${status.tools_available.join(' • ')}
+                            </div>
                         `;
-                        console.log('⚠️ System in limited mode');
                     }
                 } catch (error) {
-                    console.error('❌ Status check failed:', error);
                     const statusDiv = document.getElementById('status');
-                    statusDiv.className = 'status disabled';
-                    statusDiv.innerHTML = '<span>❌ Connection Error</span>';
+                    statusDiv.className = 'status offline';
+                    statusDiv.innerHTML = '<span>Connection error</span>';
                 }
             }
-            
+
             async function sendQuery() {
                 const input = document.getElementById('queryInput');
                 const sendBtn = document.getElementById('sendBtn');
                 const query = input.value.trim();
-                
-                if (!query) {
-                    input.focus();
-                    return;
-                }
-                
-                console.log('📤 Sending query:', query);
-                
-                // Add user message
+
+                if (!query) return;
+
                 addMessage('user', query);
                 input.value = '';
-                
-                // Show loading
+
                 sendBtn.disabled = true;
                 sendBtn.innerHTML = '<span class="loading">Processing</span>';
-                
+
                 try {
                     const response = await fetch('/query', {
                         method: 'POST',
                         headers: { 'Content-Type': 'application/json' },
                         body: JSON.stringify({ query, chat_history: chatHistory })
                     });
-                    
+
                     const result = await response.json();
-                    console.log('📥 Response received:', result);
-                    
+
                     if (result.success) {
-                        addMessage('assistant', result.response, result.sources);
-                        console.log('✅ Query processed successfully');
+                        addMessage('assistant', result.response, result.sources, result.visualizations);
                     } else {
-                        addMessage('assistant', result.response || 'An error occurred');
-                        console.log('⚠️ Query failed:', result.error);
+                        addMessage('assistant', result.response || 'Analysis failed. Please try again.');
                     }
                 } catch (error) {
-                    console.error('❌ Network error:', error);
-                    addMessage('assistant', '❌ Network error. Please check your connection and try again.');
+                    addMessage('assistant', 'Connection error. Please check your network and try again.');
                 } finally {
                     sendBtn.disabled = false;
-                    sendBtn.innerHTML = '🚀 Research';
+                    sendBtn.innerHTML = 'Research';
                     input.focus();
                 }
             }
-            
-            function addMessage(sender, content, sources = []) {
-                console.log(`💬 Adding ${sender} message`);
+
+            function addMessage(sender, content, sources = [], visualizations = []) {
                 const messagesDiv = document.getElementById('chatMessages');
+                
+                // Clear welcome message
+                if (messageCount === 0) {
+                    messagesDiv.innerHTML = '';
+                }
+                messageCount++;
+
                 const messageDiv = document.createElement('div');
                 messageDiv.className = `message ${sender}`;
-                
+
                 let sourcesHtml = '';
                 if (sources && sources.length > 0) {
-                    sourcesHtml = `<div class="sources"><span class="label">Sources:</span> ${sources.map(s => `<span>${s}</span>`).join('')}</div>`;
+                    sourcesHtml = `
+                        <div class="sources">
+                            Sources: ${sources.map(s => `<span>${s}</span>`).join('')}
+                        </div>
+                    `;
                 }
-                
-                const senderIcon = sender === 'user' ? '👤' : '🤖';
-                const senderName = sender === 'user' ? 'You' : 'AI Research Assistant';
-                
+
+                let visualizationHtml = '';
+                if (visualizations && visualizations.length > 0) {
+                    visualizationHtml = visualizations.map(viz => 
+                        `<div class="visualization-container">${viz}</div>`
+                    ).join('');
+                }
+
                 messageDiv.innerHTML = `
-                    <div class="sender">${senderIcon} ${senderName}</div>
-                    <div class="content">${content.replace(/\n/g, '<br>')}</div>
-                    ${sourcesHtml}
+                    <div class="message-content">
+                        ${content.replace(/\n/g, '<br>')}
+                        ${sourcesHtml}
+                    </div>
+                    ${visualizationHtml}
+                    <div class="message-meta">${new Date().toLocaleTimeString()}</div>
                 `;
-                
+
                 messagesDiv.appendChild(messageDiv);
                 messagesDiv.scrollTop = messagesDiv.scrollHeight;
-                
-                // Update chat history
+
                 chatHistory.push({ role: sender, content });
                 if (chatHistory.length > 20) chatHistory = chatHistory.slice(-20);
             }
-            
+
             function setQuery(query) {
-                console.log('📝 Setting query:', query);
-                const input = document.getElementById('queryInput');
-                input.value = query;
-                input.focus();
-                
-                // Optional: auto-send after a short delay
-                setTimeout(() => {
-                    if (input.value === query) { // Only if user didn't change it
-                        sendQuery();
-                    }
-                }, 100);
+                document.getElementById('queryInput').value = query;
+                setTimeout(() => sendQuery(), 100);
             }
-            
-            // Handle Enter key
-            document.getElementById('queryInput').addEventListener('keypress', function(e) {
-                if (e.key === 'Enter') {
-                    sendQuery();
-                }
+
+            // Event listeners
+            document.getElementById('queryInput').addEventListener('keypress', (e) => {
+                if (e.key === 'Enter') sendQuery();
             });
-            
+
+            document.getElementById('sendBtn').addEventListener('click', sendQuery);
+
             // Initialize
-            document.addEventListener('DOMContentLoaded', function() {
-                console.log('🚀 Web3 Research Co-Pilot initialized');
+            document.addEventListener('DOMContentLoaded', () => {
                 checkStatus();
                 document.getElementById('queryInput').focus();
             });
@@ -568,35 +777,33 @@ async def get_homepage(request: Request):
 
 @app.get("/status")
 async def get_status():
-    logger.info("📊 Status endpoint called")
+    """System status endpoint"""
     status = {
         "enabled": service.enabled,
         "gemini_configured": bool(config.GEMINI_API_KEY),
-        "tools_available": ["CoinGecko", "DeFiLlama", "Etherscan"],
+        "tools_available": ["Market Data", "DeFi Analytics", "Network Metrics"],
         "airaa_enabled": service.airaa.enabled if service.airaa else False,
-        "timestamp": datetime.now().isoformat()
+        "timestamp": datetime.now().isoformat(),
+        "version": "2.0.0"
     }
-    logger.info(f"📊 Status response: {status}")
     return status
 
 @app.post("/query", response_model=QueryResponse)
 async def process_query(request: QueryRequest):
-    logger.info(f"📥 Query endpoint called: {request.query[:50]}{'...' if len(request.query) > 50 else ''}")
-    result = await service.process_query(request.query)
-    logger.info(f"📤 Query response: success={result.success}")
-    return result
+    """Process research query"""
+    return await service.process_query(request.query)
 
 @app.get("/health")
 async def health_check():
-    logger.info("❤️ Health check endpoint called")
+    """Health check endpoint"""
     return {
         "status": "healthy",
         "timestamp": datetime.now().isoformat(),
         "service_enabled": service.enabled,
-        "version": "1.0.0"
+        "version": "2.0.0"
     }
 
 if __name__ == "__main__":
     import uvicorn
-    logger.info("🌟 Starting FastAPI server...")
+    logger.info("Starting Web3 Research Co-Pilot...")
     uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")

requirements.txt

@@ -10,3 +10,7 @@ google-generativeai
 asyncio-throttle
 fastapi
 uvicorn
+plotly
+pandas
+numpy
+jinja2

src/tools/etherscan_tool.py

@@ -12,6 +12,7 @@ class EtherscanTool(BaseWeb3Tool):
     Useful for: transaction analysis, address information, gas prices, token data.
     Input: Ethereum address, transaction hash, or general blockchain query."""
     args_schema: type[BaseModel] = Web3ToolInput
+    enabled: bool = True  # Add enabled as a Pydantic field
     
     _base_url: str = PrivateAttr(default="https://api.etherscan.io/api")
     _api_key: Optional[str] = PrivateAttr(default=None)

src/visualizations.py

@@ -0,0 +1,193 @@
+import plotly.graph_objects as go
+import plotly.express as px
+from plotly.subplots import make_subplots
+import pandas as pd
+from typing import Dict, List, Any, Optional
+import logging
+
+logger = logging.getLogger(__name__)
+
+class CryptoVisualizations:
+    """Professional cryptocurrency data visualizations"""
+    
+    @staticmethod
+    def create_price_chart(data: Dict[str, Any], symbol: str = "BTC") -> go.Figure:
+        """Create a professional price chart with volume"""
+        try:
+            if not data or 'prices' not in data:
+                return CryptoVisualizations._create_empty_chart("No price data available")
+            
+            prices = data['prices']
+            volumes = data.get('total_volumes', [])
+            
+            # Convert to DataFrame
+            df = pd.DataFrame({
+                'timestamp': [pd.to_datetime(p[0], unit='ms') for p in prices],
+                'price': [p[1] for p in prices],
+                'volume': [v[1] if v else 0 for v in (volumes[:len(prices)] if volumes else [])]
+            })
+            
+            # Create subplot with secondary y-axis
+            fig = make_subplots(
+                rows=2, cols=1,
+                shared_xaxes=True,
+                vertical_spacing=0.1,
+                subplot_titles=(f'{symbol.upper()} Price', 'Volume'),
+                row_heights=[0.7, 0.3]
+            )
+            
+            # Price line
+            fig.add_trace(
+                go.Scatter(
+                    x=df['timestamp'],
+                    y=df['price'],
+                    mode='lines',
+                    name='Price',
+                    line=dict(color='#00d4aa', width=2),
+                    hovertemplate='<b>%{y:,.2f} USD</b><br>%{x}<extra></extra>'
+                ),
+                row=1, col=1
+            )
+            
+            # Volume bars
+            if not df['volume'].empty and df['volume'].sum() > 0:
+                fig.add_trace(
+                    go.Bar(
+                        x=df['timestamp'],
+                        y=df['volume'],
+                        name='Volume',
+                        marker_color='rgba(0, 212, 170, 0.3)',
+                        hovertemplate='<b>%{y:,.0f}</b><br>%{x}<extra></extra>'
+                    ),
+                    row=2, col=1
+                )
+            
+            # Update layout
+            fig.update_layout(
+                title=dict(
+                    text=f'{symbol.upper()} Price Analysis',
+                    font=dict(size=24, color='#2c3e50'),
+                    x=0.5
+                ),
+                showlegend=False,
+                height=600,
+                margin=dict(l=60, r=30, t=80, b=60),
+                plot_bgcolor='white',
+                paper_bgcolor='white',
+                font=dict(family="SF Pro Display, -apple-system, system-ui, sans-serif", size=12)
+            )
+            
+            # Update axes
+            fig.update_xaxes(
+                gridcolor='#ecf0f1',
+                gridwidth=1,
+                showgrid=True,
+                tickfont=dict(color='#7f8c8d')
+            )
+            fig.update_yaxes(
+                gridcolor='#ecf0f1',
+                gridwidth=1,
+                showgrid=True,
+                tickfont=dict(color='#7f8c8d')
+            )
+            
+            return fig
+            
+        except Exception as e:
+            logger.error(f"Error creating price chart: {e}")
+            return CryptoVisualizations._create_empty_chart(f"Error: {str(e)}")
+    
+    @staticmethod
+    def create_market_overview(data: List[Dict[str, Any]]) -> go.Figure:
+        """Create market overview with top cryptocurrencies"""
+        try:
+            if not data:
+                return CryptoVisualizations._create_empty_chart("No market data available")
+            
+            # Convert to DataFrame
+            df = pd.DataFrame(data)
+            
+            # Take top 10 by market cap
+            df = df.head(10).sort_values('market_cap', ascending=True)
+            
+            # Create horizontal bar chart
+            fig = go.Figure()
+            
+            # Market cap bars
+            fig.add_trace(
+                go.Bar(
+                    y=df['name'],
+                    x=df['market_cap'],
+                    orientation='h',
+                    marker=dict(
+                        color=df['price_change_percentage_24h'],
+                        colorscale='RdYlGn',
+                        colorbar=dict(title="24h Change %"),
+                        line=dict(color='white', width=1)
+                    ),
+                    hovertemplate='<b>%{y}</b><br>Market Cap: $%{x:,.0f}<br>24h: %{marker.color:.2f}%<extra></extra>'
+                )
+            )
+            
+            fig.update_layout(
+                title=dict(
+                    text='Top 10 Cryptocurrencies by Market Cap',
+                    font=dict(size=24, color='#2c3e50'),
+                    x=0.5
+                ),
+                xaxis_title='Market Cap (USD)',
+                height=500,
+                margin=dict(l=120, r=30, t=80, b=60),
+                plot_bgcolor='white',
+                paper_bgcolor='white',
+                font=dict(family="SF Pro Display, -apple-system, system-ui, sans-serif", size=12)
+            )
+            
+            fig.update_xaxes(
+                gridcolor='#ecf0f1',
+                gridwidth=1,
+                showgrid=True,
+                tickfont=dict(color='#7f8c8d')
+            )
+            fig.update_yaxes(
+                tickfont=dict(color='#2c3e50', size=11)
+            )
+            
+            return fig
+            
+        except Exception as e:
+            logger.error(f"Error creating market overview: {e}")
+            return CryptoVisualizations._create_empty_chart(f"Error: {str(e)}")
+    
+    @staticmethod
+    def _create_empty_chart(message: str) -> go.Figure:
+        """Create an empty chart with error message"""
+        fig = go.Figure()
+        
+        fig.add_annotation(
+            text=message,
+            xref="paper", yref="paper",
+            x=0.5, y=0.5,
+            showarrow=False,
+            font=dict(size=16, color="#7f8c8d")
+        )
+        
+        fig.update_layout(
+            height=400,
+            margin=dict(l=60, r=60, t=60, b=60),
+            plot_bgcolor='white',
+            paper_bgcolor='white',
+            xaxis=dict(showgrid=False, showticklabels=False),
+            yaxis=dict(showgrid=False, showticklabels=False)
+        )
+        
+        return fig
+
+# Convenience functions for backward compatibility
+def create_price_chart(data: Dict[str, Any], symbol: str = "BTC") -> go.Figure:
+    """Create price chart - backward compatibility function"""
+    return CryptoVisualizations.create_price_chart(data, symbol)
+
+def create_market_overview(data: List[Dict[str, Any]]) -> go.Figure:
+    """Create market overview - backward compatibility function"""
+    return CryptoVisualizations.create_market_overview(data)

test_suite.py

@@ -0,0 +1,259 @@
+#!/usr/bin/env python3
+"""
+Comprehensive test suite for Web3 Research Co-Pilot
+"""
+
+import sys
+import asyncio
+import time
+from datetime import datetime
+
+def test_imports():
+    """Test all critical imports"""
+    print("🧪 Testing imports...")
+    
+    try:
+        # Core imports
+        from src.visualizations import CryptoVisualizations, create_price_chart
+        from src.agent.research_agent import Web3ResearchAgent
+        from src.utils.config import config
+        from src.tools.coingecko_tool import CoinGeckoTool
+        from src.tools.defillama_tool import DeFiLlamaTool
+        from src.tools.etherscan_tool import EtherscanTool
+        from src.api.airaa_integration import AIRAAIntegration
+        
+        # FastAPI app
+        from app import app, service, Web3CoPilotService
+        
+        print("✅ All imports successful")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Import failed: {e}")
+        return False
+
+def test_configuration():
+    """Test configuration setup"""
+    print("🧪 Testing configuration...")
+    
+    try:
+        from src.utils.config import config
+        
+        print(f"   • GEMINI_API_KEY: {'✅ Set' if config.GEMINI_API_KEY else '❌ Not set'}")
+        print(f"   • COINGECKO_API_KEY: {'✅ Set' if config.COINGECKO_API_KEY else '⚠️ Not set'}")
+        print(f"   • ETHERSCAN_API_KEY: {'✅ Set' if config.ETHERSCAN_API_KEY else '⚠️ Not set'}")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Configuration test failed: {e}")
+        return False
+
+def test_visualizations():
+    """Test visualization creation"""
+    print("🧪 Testing visualizations...")
+    
+    try:
+        from src.visualizations import CryptoVisualizations
+        
+        # Test empty chart
+        fig1 = CryptoVisualizations._create_empty_chart("Test message")
+        print("   ✅ Empty chart creation")
+        
+        # Test price chart with sample data
+        sample_data = {
+            'prices': [
+                [1672531200000, 16500.50], 
+                [1672617600000, 16750.25],
+                [1672704000000, 17100.00]
+            ],
+            'total_volumes': [
+                [1672531200000, 1000000], 
+                [1672617600000, 1200000],
+                [1672704000000, 1100000]
+            ]
+        }
+        
+        fig2 = CryptoVisualizations.create_price_chart(sample_data, 'BTC')
+        print("   ✅ Price chart with data")
+        
+        # Test market overview
+        market_data = [
+            {'name': 'Bitcoin', 'market_cap': 500000000000, 'price_change_percentage_24h': 2.5},
+            {'name': 'Ethereum', 'market_cap': 200000000000, 'price_change_percentage_24h': -1.2}
+        ]
+        
+        fig3 = CryptoVisualizations.create_market_overview(market_data)
+        print("   ✅ Market overview chart")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Visualization test failed: {e}")
+        return False
+
+def test_tools():
+    """Test individual tools"""
+    print("🧪 Testing tools...")
+    
+    try:
+        from src.tools.coingecko_tool import CoinGeckoTool
+        from src.tools.defillama_tool import DeFiLlamaTool
+        from src.tools.etherscan_tool import EtherscanTool
+        
+        # Test tool initialization
+        coingecko = CoinGeckoTool()
+        print("   ✅ CoinGecko tool initialization")
+        
+        defillama = DeFiLlamaTool()
+        print("   ✅ DeFiLlama tool initialization")
+        
+        etherscan = EtherscanTool()
+        print("   ✅ Etherscan tool initialization")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Tools test failed: {e}")
+        return False
+
+async def test_service():
+    """Test service functionality"""
+    print("🧪 Testing service...")
+    
+    try:
+        from app import service
+        
+        print(f"   • Service enabled: {'✅' if service.enabled else '❌'}")
+        print(f"   • Agent available: {'✅' if service.agent else '❌'}")
+        print(f"   • AIRAA enabled: {'✅' if service.airaa and service.airaa.enabled else '❌'}")
+        
+        # Test a simple query
+        if service.enabled:
+            print("   🔄 Testing query processing...")
+            response = await service.process_query("What is Bitcoin?")
+            
+            if response.success:
+                print("   ✅ Query processing successful")
+                print(f"      Response length: {len(response.response)} characters")
+            else:
+                print(f"   ⚠️ Query failed: {response.error}")
+        else:
+            print("   ⚠️ Service disabled - limited testing")
+            
+        return True
+        
+    except Exception as e:
+        print(f"❌ Service test failed: {e}")
+        return False
+
+def test_app_health():
+    """Test FastAPI app health"""
+    print("🧪 Testing FastAPI app...")
+    
+    try:
+        from fastapi.testclient import TestClient
+        from app import app
+        
+        with TestClient(app) as client:
+            # Test health endpoint
+            response = client.get("/health")
+            if response.status_code == 200:
+                print("   ✅ Health endpoint")
+            else:
+                print(f"   ❌ Health endpoint failed: {response.status_code}")
+                
+            # Test status endpoint  
+            response = client.get("/status")
+            if response.status_code == 200:
+                print("   ✅ Status endpoint")
+                status_data = response.json()
+                print(f"      Version: {status_data.get('version', 'Unknown')}")
+            else:
+                print(f"   ❌ Status endpoint failed: {response.status_code}")
+                
+            # Test homepage
+            response = client.get("/")
+            if response.status_code == 200:
+                print("   ✅ Homepage endpoint")
+            else:
+                print(f"   ❌ Homepage failed: {response.status_code}")
+            
+        return True
+        
+    except Exception as e:
+        print(f"❌ FastAPI test failed: {e}")
+        return False
+
+def run_performance_test():
+    """Simple performance test"""
+    print("🧪 Performance test...")
+    
+    try:
+        from src.visualizations import CryptoVisualizations
+        
+        # Time visualization creation
+        start_time = time.time()
+        
+        for i in range(10):
+            sample_data = {
+                'prices': [[1672531200000 + i*3600000, 16500 + i*10] for i in range(100)],
+                'total_volumes': [[1672531200000 + i*3600000, 1000000 + i*1000] for i in range(100)]
+            }
+            fig = CryptoVisualizations.create_price_chart(sample_data, 'TEST')
+        
+        end_time = time.time()
+        avg_time = (end_time - start_time) / 10
+        
+        print(f"   ⏱️ Average chart creation: {avg_time:.3f}s")
+        
+        if avg_time < 1.0:
+            print("   ✅ Performance acceptable")
+            return True
+        else:
+            print("   ⚠️ Performance slow")
+            return True
+            
+    except Exception as e:
+        print(f"❌ Performance test failed: {e}")
+        return False
+
+async def main():
+    """Run all tests"""
+    print("=" * 50)
+    print("🚀 Web3 Research Co-Pilot - Test Suite")
+    print("=" * 50)
+    print()
+    
+    test_results = []
+    
+    # Run all tests
+    test_results.append(test_imports())
+    test_results.append(test_configuration())
+    test_results.append(test_visualizations())
+    test_results.append(test_tools())
+    test_results.append(await test_service())
+    test_results.append(test_app_health())
+    test_results.append(run_performance_test())
+    
+    print()
+    print("=" * 50)
+    print("📊 Test Results Summary")
+    print("=" * 50)
+    
+    passed = sum(test_results)
+    total = len(test_results)
+    
+    print(f"Tests passed: {passed}/{total}")
+    print(f"Success rate: {(passed/total)*100:.1f}%")
+    
+    if passed == total:
+        print("🎉 All tests passed!")
+        return 0
+    else:
+        print("⚠️ Some tests failed")
+        return 1
+
+if __name__ == "__main__":
+    exit_code = asyncio.run(main())
+    sys.exit(exit_code)