Kims12 commited on
Commit
fdaadbb
ยท
verified ยท
1 Parent(s): c1f6fb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -11
app.py CHANGED
@@ -25,15 +25,13 @@ def initialize_gemini():
25
  logger.error("GEMINI_API_KEY๊ฐ€ ์„ค์ •๋˜์ง€ ์•Š์•˜์Šต๋‹ˆ๋‹ค.")
26
  return None
27
 
28
- genai.configure(api_key=api_key)
29
- return genai.GenerativeModel(
30
- model_name="gemini-2.0-flash",
31
- generation_config={
32
- "temperature": 0.2, # ๋‚ฎ์€ temperature๋กœ ์ •ํ™•ํ•œ ๋ฒˆ์—ญ ์œ ๋„
33
- "max_output_tokens": 1024,
34
- "top_p": 0.9,
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
- response = model.generate_content(translation_prompt)
52
- english_prompt = response.text.strip()
 
 
 
 
 
 
 
 
 
 
 
 
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: