Update app.py

app.py

@@ -47,6 +47,162 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+@dataclass
+class ChatContext:
+    """Track chat context and discussed NFTs."""
+    wallet_data: Dict[str, Any] = field(default_factory=dict)
+    discussed_nfts: Set[str] = field(default_factory=set)
+    last_collection: str = ""
+    
+    def mark_nft_discussed(self, contract: str, token_id: str) -> None:
+        """Track which NFTs have been discussed."""
+        self.discussed_nfts.add(f"{contract}_{token_id}")
+        
+    def is_nft_discussed(self, contract: str, token_id: str) -> bool:
+        """Check if NFT has been discussed."""
+        return f"{contract}_{token_id}" in self.discussed_nfts
+
+class BlockchainError(Exception):
+    """Base exception for blockchain-related errors."""
+    pass
+
+class ConfigError(BlockchainError):
+    """Configuration error."""
+    pass
+
+class APIError(BlockchainError):
+    """API call error."""
+    pass
+
+class ValidationError(BlockchainError):
+    """Input validation error."""
+    pass
+
+class NFTError(BlockchainError):
+    """NFT processing error."""
+    pass
+
+@dataclass
+class NFTMetadata:
+    """Structure for NFT metadata."""
+    contract: str
+    token_id: str
+    name: str
+    collection_name: str
+    image_url: Optional[str] = None
+    description: Optional[str] = None
+    traits: List[Dict[str, Any]] = field(default_factory=list)
+    last_updated: datetime = field(default_factory=datetime.now)
+    
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary format."""
+        return {
+            "contract": self.contract,
+            "token_id": self.token_id,
+            "name": self.name,
+            "collection_name": self.collection_name,
+            "image_url": self.image_url,
+            "description": self.description,
+            "traits": self.traits,
+            "last_updated": self.last_updated.isoformat()
+        }
+
+@dataclass
+class Config:
+    """Enhanced application configuration."""
+    
+    # Improved system prompt for more natural responses
+    SYSTEM_PROMPT: str = """
+    You are LOSS DOG 🐕 (Learning & Observing Smart Systems Digital Output Generator), 
+    a friendly blockchain analysis assistant!
+
+    Your conversation style:
+    - Maintain natural dialogue flow
+    - Avoid repetitive responses
+    - Use context from previous messages
+    - Track which NFTs have been discussed
+    - Provide insights when relevant
+
+    When analyzing wallets:
+    - Reference specific NFTs by collection and ID
+    - Highlight interesting patterns
+    - Compare wallets if multiple available
+    - Monitor NFT rarity and traits
+    """
+    
+    # API Configuration
+    ETHERSCAN_BASE_URL: str = "https://api.etherscan.io/api"
+    OPENSEA_BASE_URL: str = "https://api.opensea.io/api/v2"
+    ETHEREUM_ADDRESS_REGEX: str = r"^0x[a-fA-F0-9]{40}$"
+
+    # Rate Limiting & Retries
+    RATE_LIMIT_DELAY: float = 0.2  # 5 requests per second
+    MAX_RETRIES: int = 3
+    RETRY_BASE_DELAY: float = 1.0
+    RETRY_MAX_DELAY: float = 10.0
+
+    # API Timeouts
+    REQUEST_TIMEOUT: float = 30.0
+    NFT_FETCH_TIMEOUT: int = 10
+    
+    # NFT Processing
+    MAX_NFTS_PER_COLLECTION: int = 50
+    NFT_BATCH_SIZE: int = 5
+    MAX_IMAGE_SIZE: Tuple[int, int] = (800, 800)
+    
+    # OpenAI Settings
+    OPENAI_MODEL: str = "gpt-4-0125-preview"
+    MAX_TOKENS: int = 8000
+    TEMPERATURE: float = 0.7
+    HISTORY_LIMIT: int = 10"""
+Blockchain Wallet Analyzer - A tool for analyzing Ethereum wallet contents and NFT holdings.
+
+This module provides a complete implementation of a blockchain wallet analysis tool
+with a Gradio web interface. It includes improved NFT tracking, natural AI communication,
+and enhanced error handling.
+
+Author: Claude
+Date: January 2025
+"""
+
+from __future__ import annotations
+import os
+import re
+import json
+import time
+import logging
+import asyncio
+import base64
+from typing import List, Dict, Tuple, Any, Optional, TypeVar, Set
+from datetime import datetime
+from decimal import Decimal
+from dataclasses import dataclass, field
+from pathlib import Path
+from io import BytesIO
+
+import aiohttp
+import openai
+import gradio as gr
+from PIL import Image
+from tenacity import retry, stop_after_attempt, wait_exponential
+
+# Type variables
+T = TypeVar('T')
+WalletData = Dict[str, Any]
+ChatHistory = List[Tuple[str, str]]
+NFTIdentifier = Tuple[str, str]  # (contract, token_id)
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler('blockchain_analyzer.log'),
+        logging.StreamHandler()
+    ]
+)
+logger = logging.getLogger(__name__)
+
 class BlockchainError(Exception):
     """Base exception for blockchain-related errors."""
     pass
@@ -136,7 +292,7 @@ class Config:
     MAX_IMAGE_SIZE: Tuple[int, int] = (800, 800)
     
     # OpenAI Settings
-    OPENAI_MODEL: str = "gpt-4o-mini"
+    OPENAI_MODEL: str = "gpt-4-0125-preview"
     MAX_TOKENS: int = 8000
     TEMPERATURE: float = 0.7
     HISTORY_LIMIT: int = 10