Update Image_text_generation.py

Image_text_generation.py

@@ -10,9 +10,10 @@ import httpx
 genai_client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
 class Image_text_Generator:
-    def __init__(self):
-        pass
-
+    def __init__(self, user_id):
+        
+        sellf.user_id = user_id
+        
     def generate_image_with_gemini(self, prompt):
         """
         使用 Gemini 模型生成圖片。
@@ -36,7 +37,7 @@ class Image_text_Generator:
                 return part.inline_data.data
         return None
 
-    def upload_image_to_tmp(self, user_id):
+    def upload_image_to_tmp(self):
         """
         返回:
         str: 上傳後的圖片 URL，如果上傳失敗則返回 None。
@@ -46,10 +47,10 @@ class Image_text_Generator:
             image = PIL.Image.open(io.BytesIO(image_binary))
 
             # 建立暫存檔案以便上傳
-            image.save('tmp/'+user_id+'.png', format='PNG')
+            image.save('tmp/'+self.user_id+'.png', format='PNG')
 
             # 返回圖片的連結
-            return 'tmp/'+user_id+'.png'
+            return 'tmp/'+self.user_id+'.png'
         except Exception as e:
             print(f"圖片上傳失敗: {e}")
             return None