Update app.py
Browse files
app.py
CHANGED
@@ -25,15 +25,13 @@ def initialize_gemini():
|
|
25 |
logger.error("GEMINI_API_KEY๊ฐ ์ค์ ๋์ง ์์์ต๋๋ค.")
|
26 |
return None
|
27 |
|
28 |
-
genai
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
}
|
36 |
-
)
|
37 |
|
38 |
# LLM ๋ชจ๋ธ๋ก ํ๊ตญ์ด ํ๋กฌํํธ๋ฅผ ์์ด๋ก ๋ฒ์ญ
|
39 |
def translate_to_english(korean_prompt, model):
|
@@ -48,8 +46,20 @@ def translate_to_english(korean_prompt, model):
|
|
48 |
๋ฒ์ญ๋ง ์ ๊ณตํ๊ณ ๋ค๋ฅธ ์ค๋ช
์ ํ์ง ๋ง์ธ์.
|
49 |
"""
|
50 |
|
51 |
-
|
52 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
logger.info(f"๋ฒ์ญ ๊ฒฐ๊ณผ: {english_prompt}")
|
54 |
return english_prompt
|
55 |
except Exception as e:
|
|
|
25 |
logger.error("GEMINI_API_KEY๊ฐ ์ค์ ๋์ง ์์์ต๋๋ค.")
|
26 |
return None
|
27 |
|
28 |
+
# genai ํด๋ผ์ด์ธํธ ์์ฑ
|
29 |
+
client = genai.Client(api_key=api_key)
|
30 |
+
|
31 |
+
# GenerativeModel ์ธ์คํด์ค ์์ฑ
|
32 |
+
model = client.models.get_model("gemini-2.0-flash")
|
33 |
+
|
34 |
+
return model
|
|
|
|
|
35 |
|
36 |
# LLM ๋ชจ๋ธ๋ก ํ๊ตญ์ด ํ๋กฌํํธ๋ฅผ ์์ด๋ก ๋ฒ์ญ
|
37 |
def translate_to_english(korean_prompt, model):
|
|
|
46 |
๋ฒ์ญ๋ง ์ ๊ณตํ๊ณ ๋ค๋ฅธ ์ค๋ช
์ ํ์ง ๋ง์ธ์.
|
47 |
"""
|
48 |
|
49 |
+
# ์๋ก์ด API ํ์์ ๋ง๊ฒ ์์
|
50 |
+
response = model.generate_content(
|
51 |
+
types.GenerateContentRequest(
|
52 |
+
contents=[types.Content(parts=[types.Part(text=translation_prompt)])],
|
53 |
+
generation_config=types.GenerationConfig(
|
54 |
+
temperature=0.2,
|
55 |
+
max_output_tokens=1024,
|
56 |
+
top_p=0.9,
|
57 |
+
)
|
58 |
+
)
|
59 |
+
)
|
60 |
+
|
61 |
+
# ์๋ต์์ ํ
์คํธ ์ถ์ถ
|
62 |
+
english_prompt = response.candidates[0].content.parts[0].text.strip()
|
63 |
logger.info(f"๋ฒ์ญ ๊ฒฐ๊ณผ: {english_prompt}")
|
64 |
return english_prompt
|
65 |
except Exception as e:
|