Update app.py

app.py

@@ -1,14 +1,9 @@
 import gradio as gr
 import anthropic
-import os
 from typing import List, Dict, Optional, Tuple
-from http import HTTPStatus
+import json
 
 # Constants
-MODEL_NAME = "claude-3-5-sonnet-20241022"  # Updated model name
-MAX_TOKENS = 4096
-TEMPERATURE = 0.7
-
 DEFAULT_SYSTEM_PROMPT = """You are a specialized coding assistant powered by Claude 3.5 Sonnet. Your primary focus is helping with programming tasks, code review, debugging, and software architecture. You should:
 
 1. Always provide complete, production-ready code solutions
@@ -34,53 +29,35 @@ For all code examples:
 - Consider edge cases
 """
 
-class APIError(Exception):
-    """Custom exception for API-related errors"""
-    def __init__(self, message: str, status_code: Optional[int] = None):
-        super().__init__(message)
-        self.status_code = status_code
+MAX_TOKENS = 4096
+TEMPERATURE = 0.7
 
 class ChatState:
     """Manages the state of the chat session"""
     def __init__(self):
-        self.client: Optional[anthropic.Anthropic] = None
+        self.client: Optional[anthropic.Client] = None
         self.system_prompt: str = DEFAULT_SYSTEM_PROMPT
         self.history: List[Tuple[str, str]] = []
 
     def initialize_client(self, api_key: str) -> Tuple[bool, str]:
         """Initialize the Anthropic client with the provided API key"""
         try:
-            # Validate API key format
-            if not api_key.startswith('sk-'):
-                return False, "Invalid API key format. Key should start with 'sk-'"
-
-            # Initialize client with proper error handling
-            self.client = anthropic.Anthropic(
-                api_key=api_key,
-            )
-
+            self.client = anthropic.Client(api_key=api_key)
             # Test the API key with a minimal request
-            response = self.client.messages.create(
-                model=MODEL_NAME,
+            self.client.messages.create(
+                model="claude-3-sonnet-20241022",
                 max_tokens=10,
-                messages=[
-                    {"role": "user", "content": "test"}
-                ]
+                messages=[{"role": "user", "content": "test"}]
             )
-
-            return True, "✅ API key validated successfully!"
-
-        except anthropic.APIError as e:
-            self.client = None
-            if e.status_code == HTTPStatus.UNAUTHORIZED:
-                return False, "❌ Invalid API key. Please check your credentials."
-            elif e.status_code == HTTPStatus.TOO_MANY_REQUESTS:
-                return False, "❌ Rate limit exceeded. Please try again later."
-            else:
-                return False, f"❌ API Error: {str(e)}"
+            return True, "API key validated successfully!"
         except Exception as e:
             self.client = None
-            return False, f"❌ Error: {str(e)}"
+            return False, f"Error validating API key: {str(e)}"
+
+    def update_system_prompt(self, new_prompt: str) -> str:
+        """Update the system prompt"""
+        self.system_prompt = new_prompt
+        return "System prompt updated successfully!"
 
     def generate_response(self, message: str) -> str:
         """Generate a response using the Claude API"""
@@ -88,22 +65,18 @@ class ChatState:
             return "Please enter a valid API key first."
 
         try:
-            # Construct messages including system prompt and history
             messages = [{"role": "system", "content": self.system_prompt}]
             
-            # Add conversation history
             for human_msg, assistant_msg in self.history:
                 messages.extend([
                     {"role": "user", "content": human_msg},
                     {"role": "assistant", "content": assistant_msg}
                 ])
             
-            # Add current message
             messages.append({"role": "user", "content": message})
 
-            # Make API call with all recommended parameters
             response = self.client.messages.create(
-                model=MODEL_NAME,
+                model="claude-3-sonnet-20241022",
                 max_tokens=MAX_TOKENS,
                 temperature=TEMPERATURE,
                 messages=messages
@@ -111,10 +84,6 @@ class ChatState:
 
             return response.content[0].text
 
-        except anthropic.APIError as e:
-            if e.status_code == HTTPStatus.TOO_MANY_REQUESTS:
-                return "Rate limit exceeded. Please try again in a few seconds."
-            return f"API Error: {str(e)}"
         except Exception as e:
             return f"Error generating response: {str(e)}"
 
@@ -127,62 +96,71 @@ def create_demo() -> gr.Blocks:
     chat_state = ChatState()
 
     with gr.Blocks(title="Claude Coding Assistant", theme=gr.themes.Soft()) as demo:
-        gr.Markdown("""
-        # Claude Coding Assistant 🤖💻
-        
-        A specialized coding assistant powered by Claude 3.5 Sonnet. Features:
-        - Complete code solutions with best practices
-        - Detailed explanations and documentation
-        - Code review and optimization suggestions
-        - Security considerations and design patterns
-        
-        Enter your Anthropic API key (starts with 'sk-') to begin.
-        """)
-
-        # API Key input section
         with gr.Row():
-            api_key_input = gr.Textbox(
-                label="Anthropic API Key",
-                placeholder="sk-...",
-                type="password",
-                scale=4
-            )
-            validate_btn = gr.Button("🔑 Validate API Key", scale=1)
-            status_text = gr.Textbox(
-                label="Status",
-                interactive=False,
-                scale=2
-            )
-
-        # Chat interface
-        chatbot = gr.Chatbot(
-            height=600,
-            show_label=False,
-            container=True,
-            show_copy_button=True,
-            bubble_full_width=False
-        )
-        
-        with gr.Row():
-            message_input = gr.Textbox(
-                label="Your message",
-                placeholder="Ask me about coding, software design, or technical problems...",
-                lines=3,
-                scale=4
-            )
-            submit_btn = gr.Button("🚀 Send", scale=1)
-            clear_btn = gr.Button("🗑️ Clear Chat", scale=1)
+            # Main chat area (2/3 width)
+            with gr.Column(scale=2):
+                gr.Markdown("""
+                # Claude Coding Assistant 🤖💻
+                
+                A specialized coding assistant powered by Claude 3.5 Sonnet. Features:
+                - Complete code solutions with best practices
+                - Detailed explanations and documentation
+                - Code review and optimization suggestions
+                - Security considerations and design patterns
+                """)
+
+                # API Key input section
+                with gr.Row():
+                    api_key_input = gr.Textbox(
+                        label="Anthropic API Key",
+                        placeholder="Enter your API key...",
+                        type="password",
+                        scale=4
+                    )
+                    validate_btn = gr.Button("Validate API Key", scale=1)
+                    status_text = gr.Textbox(label="Status", interactive=False, scale=2)
+
+                # Chat interface
+                chatbot = gr.Chatbot(
+                    height=600,
+                    show_label=False,
+                    container=True,
+                    show_copy_button=True
+                )
+                
+                with gr.Row():
+                    message_input = gr.Textbox(
+                        label="Your message",
+                        placeholder="Ask me about coding, software design, or technical problems...",
+                        lines=3,
+                        scale=4
+                    )
+                    submit_btn = gr.Button("Send", scale=1)
+                    clear_btn = gr.Button("Clear Chat", scale=1)
+
+            # Sidebar for system prompt (1/3 width)
+            with gr.Column(scale=1):
+                gr.Markdown("### System Prompt Editor")
+                system_prompt_input = gr.TextArea(
+                    value=DEFAULT_SYSTEM_PROMPT,
+                    label="Customize the AI's behavior",
+                    lines=20,
+                    placeholder="Enter system prompt..."
+                )
+                update_prompt_btn = gr.Button("Update System Prompt")
+                prompt_status = gr.Textbox(label="Prompt Status", interactive=False)
 
         # Event handlers
         def validate_api_key(api_key: str) -> str:
             success, message = chat_state.initialize_client(api_key)
             return message
 
+        def update_system_prompt(new_prompt: str) -> str:
+            return chat_state.update_system_prompt(new_prompt)
+
         def respond(message: str, history: List[Tuple[str, str]]) -> Tuple[List[Tuple[str, str]], str]:
             if not chat_state.client:
-                return history + [(message, "Please validate your API key first.")], ""
-            if not message.strip():
-                return history, ""
+                return history + [("", "Please validate your API key first.")], ""
             
             response = chat_state.generate_response(message)
             chat_state.history = history + [(message, response)]
@@ -199,6 +177,12 @@ def create_demo() -> gr.Blocks:
             outputs=[status_text]
         )
 
+        update_prompt_btn.click(
+            update_system_prompt,
+            inputs=[system_prompt_input],
+            outputs=[prompt_status]
+        )
+
         submit_btn.click(
             respond,
             inputs=[message_input, chatbot],
@@ -216,16 +200,16 @@ def create_demo() -> gr.Blocks:
             outputs=[chatbot, message_input]
         )
 
-        # Add coding-specific examples
+        # Add examples
         gr.Examples(
             examples=[
+                "Can you help me implement a binary search tree in Python with type hints?",
                 "Review this code for security issues:\n```python\ndef process_user_input(data):\n    result = eval(data)\n    return result```",
-                "Create a TypeScript React component that implements infinite scrolling with proper error handling and loading states.",
-                "Explain how to implement rate limiting in a distributed system using Redis, with code examples.",
-                "Write a Python function that implements merge sort with comprehensive type hints and unit tests.",
+                "Explain the SOLID principles with practical examples in TypeScript.",
+                "How would you implement rate limiting in a REST API?",
             ],
             inputs=message_input,
-            label="Example Coding Questions"
+            label="Example Questions"
         )
 
     return demo