Update app.py

app.py

@@ -191,35 +191,35 @@ class ChatState:
         return "System prompt updated successfully!"
 
     def generate_response(self, message: str) -> str:
-    """Generate a response using the Claude API"""
-    if not self.client:
-        return "Please enter a valid API key first."
-
-    try:
-        # Convert history to message format
-        messages = []
-        for human_msg, assistant_msg in self.history:
-            messages.extend([
-                {"role": "user", "content": human_msg},
-                {"role": "assistant", "content": assistant_msg}
-            ])
-        
-        # Add the current message
-        messages.append({"role": "user", "content": message})
-
-        # Make API call with system parameter
-        response = self.client.messages.create(
-            model="claude-3-5-sonnet-20241022",
-            max_tokens=MAX_TOKENS,
-            temperature=TEMPERATURE,
-            system=self.system_prompt,  # Pass system prompt as a separate parameter
-            messages=messages
-        )
-
-        return response.content[0].text
-
-    except Exception as e:
-        return f"Error generating response: {str(e)}"
+        """Generate a response using the Claude API"""
+        if not self.client:
+            return "Please enter a valid API key first."
+    
+        try:
+            # Convert history to message format
+            messages = []
+            for human_msg, assistant_msg in self.history:
+                messages.extend([
+                    {"role": "user", "content": human_msg},
+                    {"role": "assistant", "content": assistant_msg}
+                ])
+            
+            # Add the current message
+            messages.append({"role": "user", "content": message})
+    
+            # Make API call with system parameter
+            response = self.client.messages.create(
+                model="claude-3-5-sonnet-20241022",
+                max_tokens=MAX_TOKENS,
+                temperature=TEMPERATURE,
+                system=self.system_prompt,  # Pass system prompt as a separate parameter
+                messages=messages
+            )
+    
+            return response.content[0].text
+    
+        except Exception as e:
+            return f"Error generating response: {str(e)}"
 
     def clear_history(self) -> None:
         """Clear the conversation history"""