Update app.py

app.py

@@ -38,7 +38,7 @@ class CalendarService:
     def __init__(self):
         self.toolset = ComposioToolSet(api_key=os.getenv('COMPOSIO_API_KEY'))
         self.connections: Dict[str, Dict[str, Any]] = {}
-
+        self.connectionRequest = None
     def analyze_calendar_events(self, response_data):
         """
         Analyze calendar events and return statistics about meetings.
@@ -133,6 +133,8 @@ class CalendarService:
                 'redirect_url': connection_request.redirectUrl,
                 'created_at': datetime.now().isoformat()
             }
+
+            self.connectionRequest = connection_request
             
             return APIResponse(
                 success=True,
@@ -152,22 +154,20 @@ class CalendarService:
     
     def check_status(self, entity_id: str) -> APIResponse:
         try:
-            if entity_id not in self.connections:
+            status = self.connectionRequest.connectionStatus
+
+            if status == 'ACTIVE':
+                status='active'
+                connection = self.connections[entity_id]
+                
                 return APIResponse(
-                    success=False,
-                    error="No connection found for this entity ID"
+                    success=True,
+                    data={
+                        'status': status,
+                        'message': f"Connection status: {connection['status']}"
+                    }
                 )
-            
-            connection = self.connections[entity_id]
-            
-            return APIResponse(
-                success=True,
-                data={
-                    'status': connection['status'],
-                    'message': f"Connection status: {connection['status']}"
-                }
-            )
-            
+                
         except Exception as e:
             return APIResponse(
                 success=False,