Update app.py

app.py

@@ -497,7 +497,7 @@ class NFTProcessor:
         except Exception as e:
             logger.error(f"Error processing image: {e}")
             return None
-class ChatInterface:
+ class ChatInterface:
     """Manages chat interactions and context."""
     
     def __init__(self, openai_key: str, etherscan_key: str, opensea_key: str):
@@ -661,6 +661,122 @@ class ChatInterface:
             await self.nft_processor.__aexit__(None, None, None)
             self.nft_processor = None
 
+def create_gradio_interface() -> gr.Blocks:
+    """Create Gradio web interface."""
+    chat_interface = None
+
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        gr.Markdown("""
+        # 🐕 LOSS DOG: Blockchain Wallet Analyzer
+        
+        Enter your API keys and chat with your friendly blockchain-sniffing puppy!
+        - Paste an Ethereum address to analyze
+        - Chat about the wallet contents
+        - View NFT images in the gallery
+        """)
+
+        with gr.Row():
+            openai_key = gr.Textbox(
+                label="OpenAI API Key",
+                type="password",
+                placeholder="sk-..."
+            )
+            etherscan_key = gr.Textbox(
+                label="Etherscan API Key",
+                type="password",
+                placeholder="Enter Etherscan key..."
+            )
+            opensea_key = gr.Textbox(
+                label="OpenSea API Key",
+                type="password",
+                placeholder="Enter OpenSea key..."
+            )
+
+        validation_status = gr.Textbox(label="Status", interactive=False)
+        validate_btn = gr.Button("Validate Keys", variant="primary")
+
+        with gr.Row():
+            with gr.Column(scale=2):
+                chatbot = gr.Chatbot(
+                    label="Chat with LOSS DOG 🐕",
+                    height=500,
+                    bubble_full_width=False
+                )
+                with gr.Row():
+                    msg_input = gr.Textbox(
+                        label="Message",
+                        placeholder="Enter ETH address or ask about the wallet...",
+                        scale=8
+                    )
+                    send_btn = gr.Button("Send", scale=1, variant="primary")
+                    clear_btn = gr.Button("Clear", scale=1)
+
+            with gr.Column(scale=1):
+                nft_gallery = gr.Gallery(
+                    label="NFT Gallery",
+                    columns=2,
+                    height=400
+                )
+                wallet_data = gr.JSON(label="Wallet Data")
+
+        def init_interface(openai_k: str, etherscan_k: str, opensea_k: str) -> str:
+            """Initialize chat interface with API keys."""
+            nonlocal chat_interface
+            try:
+                chat_interface = ChatInterface(openai_k, etherscan_k, opensea_k)
+                return "✅ API keys validated successfully!"
+            except Exception as e:
+                return f"❌ Error: {str(e)}"
+
+        async def process_message(
+            message: str,
+            history: List[Tuple[str, str]]
+        ) -> Tuple[List[Tuple[str, str]], Dict[str, Any], List[Image.Image]]:
+            """Process user message."""
+            if not chat_interface:
+                return [], {}, []
+            return await chat_interface.process_message(message, history)
+
+        def clear_chat():
+            """Clear chat and context."""
+            if chat_interface:
+                return chat_interface.clear_context()
+            return {}, []
+
+        # Wire up the interface
+        validate_btn.click(
+            init_interface,
+            inputs=[openai_key, etherscan_key, opensea_key],
+            outputs=[validation_status]
+        )
+
+        msg_input.submit(
+            process_message,
+            inputs=[msg_input, chatbot],
+            outputs=[chatbot, wallet_data, nft_gallery]
+        ).then(
+            lambda: gr.update(value=""),
+            None,
+            [msg_input]
+        )
+
+        send_btn.click(
+            process_message,
+            inputs=[msg_input, chatbot],
+            outputs=[chatbot, wallet_data, nft_gallery]
+        ).then(
+            lambda: gr.update(value=""),
+            None,
+            [msg_input]
+        )
+
+        clear_btn.click(
+            clear_chat,
+            outputs=[wallet_data, chatbot]
+        )
+
+    return demo
+
 def main():
     """Main entry point."""
     demo = create_gradio_interface()