Update main.py

main.py

@@ -120,11 +120,12 @@ def generate_image_with_gemini(prompt):
         )
 
     for part in response.candidates[0].content.parts:
-      if part.text is not None:
-        print(part.text)
-      elif part.inline_data is not None:
-        #image = PIL.Image.open(io.BytesIO(part.inline_data.data))
-        return part.inline_data.data
+        if part.text is not None:
+            print(part.text)
+        elif part.inline_data is not None:
+          #image = PIL.Image.open(io.BytesIO(part.inline_data.data))
+          return part.inline_data.data
+
 
 # 使用 Hugging Face 生成圖片
 def generate_image_hf(prompt):