Rename Image_generation.py to Image_text_generation.py

Image_generation.py → Image_text_generation.py

@@ -5,11 +5,12 @@ from datetime import datetime
 import PIL.Image
 from google import genai
 from google.genai import types
+import httpx
 from imgurpython import ImgurClient
 
 genai_client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
-class ImageGenerator:
+class Image_text_Generator:
     def __init__(self, imgur):
         # 初始化 Imgur 客戶端
         self.imgur_client = imgur
@@ -74,3 +75,15 @@ class ImageGenerator:
         except Exception as e:
             print(f"圖片上傳失敗: {e}")
             return None
+
+    def generate_txt_with_gemini(self, prompt,doc_url):
+        doc_data = httpx.get(doc_url).content
+
+        response = client.models.generate_content(
+            model="gemini-1.5-flash",
+            contents=[
+                types.Part.from_bytes(data=doc_data, mime_type='application/pdf',),prompt]
+        )
+        
+        return response.text
+