alanchen1115 commited on
Commit
da72bf2
·
verified ·
1 Parent(s): 51e15ad

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -12
main.py CHANGED
@@ -11,7 +11,6 @@ import json, os
11
  import io
12
  import PIL.Image
13
 
14
- from imgurpython import ImgurClient
15
  from Image_text_generation import Image_text_Generator
16
  from Uploading_images_file import get_image_url,get_file_url, store_user_message, analyze_with_gemini, get_previous_message
17
 
@@ -27,18 +26,12 @@ genai_gen.configure(api_key=google_api)
27
  generation_config = genai_gen.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
28
 
29
  # 使用 gemini-2.0-flash-exp 模型
30
- model = genai_gen.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答,但如果用戶使用詢問英文問題,就用英文回應。你現在是個專業助理,職稱為OPEN小助理,個性活潑、樂觀,願意回答所有問題", generation_config=generation_config)
31
 
32
  # 設定 Line Bot 的 API 金鑰和秘密金鑰
33
  line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
34
  line_handler = WebhookHandler(os.environ["CHANNEL_SECRET"])
35
 
36
- # 設定 Imgur 的 API
37
- client_id = os.environ.get("IMGUR_CLIENT_ID")
38
- client_secret = os.environ.get("IMGUR_CLIENT_SECRET")
39
- access_token = os.environ.get("IMGUR_ACCESS_TOKEN")
40
- refresh_token = os.environ.get("IMGUR_REFRESH_TOKEN")
41
-
42
  # 設定是否正在與使用者交談
43
  working_status = os.getenv("DEFALUT_TALKING", default = "true").lower() == "true"
44
 
@@ -100,14 +93,12 @@ def handle_image_message(event):
100
 
101
  # 先立即回覆避免token過期
102
  line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中~ 請稍候.....✨"))
103
- imgur = ImgurClient(client_id, client_secret, access_token, refresh_token)
104
- image_generator = Image_text_Generator(imgur)
105
  # 生成圖片
106
  image_binary = image_generator.generate_image_with_gemini(prompt)
107
 
108
  if image_binary:
109
- album = "nvsYwgq"
110
- image_url = image_generator.upload_image_to_imgur(image_binary,album)
111
 
112
  if image_url:
113
  # 使用 push message 發送圖片,避免 reply token 超時
 
11
  import io
12
  import PIL.Image
13
 
 
14
  from Image_text_generation import Image_text_Generator
15
  from Uploading_images_file import get_image_url,get_file_url, store_user_message, analyze_with_gemini, get_previous_message
16
 
 
26
  generation_config = genai_gen.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
27
 
28
  # 使用 gemini-2.0-flash-exp 模型
29
+ model = genai_gen.GenerativeModel('gemini-2.0-flash', system_instruction="主要用繁體中文回答,但如果用戶使用詢問英文問題,就用英文回應。你現在是個專業助理,職稱為OPEN小助理,個性活潑、樂觀,願意回答所有問題", generation_config=generation_config)
30
 
31
  # 設定 Line Bot 的 API 金鑰和秘密金鑰
32
  line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
33
  line_handler = WebhookHandler(os.environ["CHANNEL_SECRET"])
34
 
 
 
 
 
 
 
35
  # 設定是否正在與使用者交談
36
  working_status = os.getenv("DEFALUT_TALKING", default = "true").lower() == "true"
37
 
 
93
 
94
  # 先立即回覆避免token過期
95
  line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中~ 請稍候.....✨"))
96
+ image_generator = Image_text_Generator(user_id)
 
97
  # 生成圖片
98
  image_binary = image_generator.generate_image_with_gemini(prompt)
99
 
100
  if image_binary:
101
+ image_url = image_generator.upload_image_to_imgur()
 
102
 
103
  if image_url:
104
  # 使用 push message 發送圖片,避免 reply token 超時