Update main.py

main.py

@@ -597,8 +597,19 @@ def approve_project_plan(project_id):
         chat = client.chats.create(model=GENERATION_MODEL, config=types.GenerateContentConfig(response_modalities=["Text", "Image"]))
         full_resp = chat.send_message([detailed_prompt, pil_image])
         
-        combined_text = "".join(part.text + "\n" for part in full_resp.parts if hasattr(part, 'text') and part.text)
-        inline_images = [Image.open(BytesIO(part.inline_data.data)) for part in full_resp.parts if hasattr(part, 'inline_data')]
+        # Fix: Access the parts through the correct structure
+        gen_parts = full_resp.candidates[0].content.parts
+        
+        combined_text = ""
+        inline_images = []
+        for part in gen_parts:
+            if part.text is not None:
+                combined_text += part.text + "\n"
+            if part.inline_data is not None:
+                img = Image.open(BytesIO(part.inline_data.data))
+                inline_images.append(img)
+        
+        combined_text = combined_text.strip()
 
         tools_section = re.search(r"TOOLS AND MATERIALS:\s*(.*?)\s*STEPS:", combined_text, re.DOTALL).group(1).strip()
         steps_section = re.search(r"STEPS:\s*(.*)", combined_text, re.DOTALL).group(1).strip()