shhuang-must commited on
Commit
428f5d6
ยท
verified ยท
1 Parent(s): caaaddb

Delete gemini.py

Browse files
Files changed (1) hide show
  1. gemini.py +0 -57
gemini.py DELETED
@@ -1,57 +0,0 @@
1
- import os
2
- from io import BytesIO
3
- from PIL import Image
4
- import gradio as gr
5
- from google import genai
6
- from google.genai import types
7
- import logging
8
-
9
- # ่จญๅฎš logging
10
- logging.basicConfig(
11
- filename='app.log',
12
- level=logging.INFO,
13
- format='%(asctime)s - %(levelname)s - %(message)s'
14
- )
15
-
16
- # ่จญๅฎš Gemini API ้‡‘้‘ฐ
17
- # === ๅˆๅง‹ๅŒ– Google Gemini ===
18
- GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
19
- client = genai.Client(api_key=GEMINI_API_KEY)
20
-
21
- def generate_image(prompt):
22
- """
23
- ไฝฟ็”จ Gemini API ๆ นๆ“šๆ็คบ่ฉž็”Ÿๆˆๅœ–็‰‡๏ผŒไธฆ่ฟ”ๅ›ž PIL ๅœ–ๅƒ็‰ฉไปถใ€‚
24
- """
25
- response = client.models.generate_content(
26
- model="gemini-2.0-flash-exp-image-generation",
27
- contents=prompt,
28
- config=types.GenerateContentConfig(
29
- response_modalities=["TEXT", "IMAGE"]
30
- ),
31
- )
32
-
33
- # ่™•็†ๅ›žๆ‡‰ไธญ็š„ๅœ–็‰‡
34
- for part in response.candidates[0].content.parts:
35
- if part.inline_data is not None:
36
- image = Image.open(BytesIO(part.inline_data.data))
37
- logging.info("ๆˆๅŠŸ็”Ÿๆˆๅœ–็‰‡ใ€‚")
38
- return image
39
-
40
- logging.warning("ๆœช่ƒฝ็”Ÿๆˆๅœ–็‰‡๏ผŒ่ซ‹ๅ˜—่ฉฆๅ…ถไป–ๆ็คบ่ฉžใ€‚")
41
- return None
42
-
43
- # ๅปบ็ซ‹ Gradio ไป‹้ข
44
- with gr.Blocks() as demo:
45
- gr.Markdown("## ๐Ÿ–ผ๏ธ Gemini ๅœ–็‰‡็”Ÿๆˆๅ™จ")
46
- prompt_input = gr.Textbox(label="่ผธๅ…ฅๆ็คบ่ฉž", placeholder="ไพ‹ๅฆ‚๏ผšไธ€้šปๆˆด่‘—ๅขจ้ก็š„่ฒ“ๅœจๆฒ™็˜ไธŠ")
47
- generate_button = gr.Button("็”Ÿๆˆๅœ–็‰‡")
48
- image_output = gr.Image(label="็”Ÿๆˆ็š„ๅœ–็‰‡")
49
-
50
- def on_generate(prompt):
51
- image = generate_image(prompt)
52
- return image
53
-
54
- generate_button.click(fn=on_generate, inputs=prompt_input, outputs=image_output)
55
-
56
- if __name__ == "__main__":
57
- demo.launch()