Update app.py

app.py

@@ -59,7 +59,6 @@ class SupplyChainOptimizer:
             print("Tavily API key not found. Search functionality will use demo mode.")
             self.tavily = None
 
-        
         # Initialize agents
         self._setup_agents()
         
@@ -68,30 +67,31 @@ class SupplyChainOptimizer:
         self.latest_optimization = ""
         self.search_results = ""
 
-     def call_claude_api(self, prompt, system_message=""):
-         """Call Claude via AWS Bedrock"""
-         if self.demo_mode:
-             return "Demo mode response"
-             try:
-                 body = {
-                     "anthropic_version": "bedrock-2023-05-31",
-                     "max_tokens": 4000,
-                     "system": system_message,
-                     "messages": [
-                         {
-                             "role": "user",
-                             "content": prompt
-                         }
-                     ]
-                 }
-                 response = self.bedrock_client.invoke_model(
-                     modelId="anthropic.claude-3-haiku-20240307-v1:0",
-                     body=json.dumps(body)
-                 )
-                 response_body = json.loads(response['body'].read())
-                 return response_body['content'][0]['text']
-             except Exception as e:
-                 return f"Error calling Claude API: {str(e)}"   
+    def call_claude_api(self, prompt, system_message=""):
+        """Call Claude via AWS Bedrock"""
+        if self.demo_mode:
+            return "Demo mode response"
+        
+        try:
+            body = {
+                "anthropic_version": "bedrock-2023-05-31",
+                "max_tokens": 4000,
+                "system": system_message,
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": prompt
+                    }
+                ]
+            }
+            response = self.bedrock_client.invoke_model(
+                modelId="anthropic.claude-3-haiku-20240307-v1:0",
+                body=json.dumps(body)
+            )
+            response_body = json.loads(response['body'].read())
+            return response_body['content'][0]['text']
+        except Exception as e:
+            return f"Error calling Claude API: {str(e)}"   
 
     def _setup_agents(self):
         """Setup agents with AWS Bedrock Claude"""