Update main.py

main.py

@@ -28,7 +28,7 @@ genai_gen.configure(api_key=os.environ["GOOGLE_API_KEY"])
 # 設定生成文字的參數
 generation_config = genai_gen.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
 
-# 使用 Gemini-1.5-flash 模型
+# 使用 gemini-2.0-flash-exp 模型
 model = genai_gen.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
@@ -107,18 +107,12 @@ def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image",
 
 # 使用 Gemini 生成圖片
 def generate_image_with_gemini(prompt):
-    
     client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
-    response = client.models.generate_content(
-                model="gemini-2.0-flash-exp-image-generation",
-                contents=prompt,
-                config=types.GenerateContentConfig(
-                response_modalities=['Text', 'Image'])
-    )
+    response = client.models.generate_content(model="gemini-2.0-flash-exp-image-generation",contents=prompt,config=types.GenerateContentConfig(response_modalities=['Text', 'Image']))
 
     for part in response.candidates[0].content.parts:
         if part.text is not None:
-            print(part.text)
+            #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