Update main.py

main.py

@@ -160,8 +160,7 @@ async def wait_for_completion(operation_id: str, max_wait_time: int = 300) -> di
         status_data = await check_generation_status(operation_id)
         
         # State 2 means completed
-        if status_data.get("state
-") == 2:
+        if status_data.get("state") == 2:
             return status_data["complete"]
         
         # Check if we've exceeded max wait time