feat: more features

README_simple.md

@@ -0,0 +1,65 @@
+---
+title: Web3 Research Co-Pilot
+emoji: 🚀
+colorFrom: blue
+colorTo: purple
+sdk: docker
+sdk_version: "latest"
+app_file: app.py
+pinned: false
+---
+
+# 🚀 Web3 Research Co-Pilot
+
+AI-powered cryptocurrency research assistant with real-time Web3 data analysis.
+
+**Live Demo**: https://archcoder-web3-copilot.hf.space
+
+## Quick Start
+
+1. **Install dependencies**:
+   ```bash
+   pip install -r requirements.txt
+   ```
+
+2. **Set up API key**:
+   ```bash
+   export GEMINI_API_KEY="your_gemini_api_key"
+   ```
+
+3. **Run**:
+   ```bash
+   python app.py
+   ```
+
+4. **Open**: http://localhost:7860
+
+## Features
+
+- 🤖 **AI Analysis** with Google Gemini
+- 📊 **Real-time Data** from CoinGecko, DeFiLlama, Etherscan
+- 📈 **Interactive Charts** and visualizations
+- 💼 **Professional UI** with FastAPI
+
+## API Keys
+
+- **Required**: [GEMINI_API_KEY](https://aistudio.google.com/) 
+- **Optional**: COINGECKO_API_KEY, ETHERSCAN_API_KEY
+
+## Deploy to HuggingFace Spaces
+
+```bash
+git remote add hf https://huggingface.co/spaces/YOUR_USERNAME/YOUR_SPACE_NAME
+git push hf main
+```
+
+## Usage Examples
+
+- "What's Bitcoin's current price?"
+- "Show top DeFi protocols by TVL"
+- "Analyze Ethereum gas prices"
+- "Compare BTC vs ETH performance"
+
+---
+
+Built with ❤️ for Web3 research

src/tools/base_tool.py

@@ -1,7 +1,7 @@
 from abc import ABC, abstractmethod
-from typing import Dict, Any, Optional
+from typing import Dict, Any, Optional, Union
 from langchain.tools import BaseTool
-from pydantic import BaseModel, Field, PrivateAttr
+from pydantic import BaseModel, Field, PrivateAttr, field_validator
 import asyncio
 import aiohttp
 import hashlib
@@ -14,7 +14,19 @@ logger = get_logger(__name__)
 
 class Web3ToolInput(BaseModel):
     query: str = Field(description="Search query or parameter")
-    filters: Optional[Dict[str, Any]] = Field(default=None, description="Additional filters")
+    filters: Optional[Union[Dict[str, Any], str]] = Field(default=None, description="Additional filters (dict) or filter type (string)")
+
+    @field_validator('filters')
+    @classmethod
+    def validate_filters(cls, v):
+        if v is None:
+            return None
+        if isinstance(v, str):
+            # Convert string filter to dict format
+            return {"type": v}
+        if isinstance(v, dict):
+            return v
+        return None
 
 class BaseWeb3Tool(BaseTool, ABC):
     name: str = "base_web3_tool"
@@ -72,33 +84,11 @@ class BaseWeb3Tool(BaseTool, ABC):
         key_data = f"{url}:{json.dumps(params, sort_keys=True)}"
         return hashlib.md5(key_data.encode()).hexdigest()[:16]
     
-    def _run(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
+    def _run(self, query: str, filters: Optional[Dict[str, Any]] = None, **kwargs) -> str:
         return asyncio.run(self._arun(query, filters))
-    
-    def run(self, tool_input: Dict[str, Any]) -> str:
-        """Handle both direct calls and LangChain tool calling format"""
-        if isinstance(tool_input, dict):
-            query = tool_input.get('query', '')
-            filters = tool_input.get('filters')
-        else:
-            # Fallback for string input
-            query = str(tool_input)
-            filters = None
-        return self._run(query, filters)
-    
-    async def arun(self, tool_input: Dict[str, Any]) -> str:
-        """Async version of run method"""
-        if isinstance(tool_input, dict):
-            query = tool_input.get('query', '')
-            filters = tool_input.get('filters')
-        else:
-            # Fallback for string input
-            query = str(tool_input)
-            filters = None
-        return await self._arun(query, filters)
 
     @abstractmethod
-    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
+    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None, **kwargs) -> str:
         pass
 
     async def cleanup(self):

src/tools/coingecko_tool.py

@@ -23,7 +23,7 @@ class CoinGeckoTool(BaseWeb3Tool):
     def __init__(self):
         super().__init__()
 
-    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
+    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None, **kwargs) -> str:
         filters = filters or {}
         try:
             # Check cache first

src/tools/defillama_tool.py

@@ -17,7 +17,7 @@ class DeFiLlamaTool(BaseWeb3Tool):
     def __init__(self):
         super().__init__()
     
-    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
+    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None, **kwargs) -> str:
         try:
             filters = filters or {}
             

src/tools/etherscan_tool.py

@@ -25,7 +25,7 @@ class EtherscanTool(BaseWeb3Tool):
         if not self.enabled:
             logger.warning("Etherscan API key not configured - limited functionality")
     
-    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
+    async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None, **kwargs) -> str:
         if not self.enabled:
             return "⚠️ **Etherscan Service Limited**\n\nEtherscan functionality requires an API key.\nGet yours free at: https://etherscan.io/apis\n\nSet environment variable: `ETHERSCAN_API_KEY=your_key`"