benjamin5607 commited on
Commit
52e1ef9
ยท
verified ยท
1 Parent(s): fbd0c37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from google import genai # ๋ฐ˜๋“œ์‹œ requirements.txt์— google-genai๊ฐ€ ์žˆ์–ด์•ผ ํ•จ
3
  import edge_tts
4
  import os
5
 
@@ -53,23 +53,23 @@ MEMBERS = {
53
 
54
  async def band_consulting(user_input, member_name):
55
  try:
56
- # ์ด๋ฆ„ ์ฒดํฌ
57
  if member_name not in MEMBERS:
58
  return f"Error: ์ด๋ฆ„ ๋ถˆ์ผ์น˜! ('{member_name}')", None
59
 
60
  member = MEMBERS[member_name]
61
 
62
- # [Step 1] ๋‡Œ: Gemini 2.0 ํ˜ธ์ถœ (์ตœ์‹  ๋ชจ๋ธ ๋ช…์‹œ!)
63
  chat_prompt = f"{member['prompt']}\n\nUser's Error/Worry: {user_input}\nResponse:"
64
 
65
  try:
66
- # โ˜… ์—ฌ๊ธฐ๊ฐ€ ํ•ต์‹ฌ์ž…๋‹ˆ๋‹ค! ๋ชจ๋ธ๋ช…์„ ์ •ํ™•ํžˆ ์ง€์ •ํ–ˆ์Šต๋‹ˆ๋‹ค.
67
  response = client.models.generate_content(
68
- model='gemini-2.0-flash-exp',
69
  contents=chat_prompt
70
  )
71
  ai_text = response.text
72
  except Exception as e:
 
73
  return f"Error (Gemini): {str(e)}", None
74
 
75
  # [Step 2] ์ž…: Edge-TTS
@@ -95,7 +95,6 @@ with gr.Blocks() as demo:
95
  out_audio = gr.Audio(label="์Œ์„ฑ")
96
  btn = gr.Button("์ƒ๋‹ด ์š”์ฒญ")
97
 
98
- # API ์ด๋ฆ„ 'predict' ์ง€์ •
99
  btn.click(fn=band_consulting, inputs=[inp_text, inp_member], outputs=[out_text, out_audio], api_name="predict")
100
 
101
  demo.launch()
 
1
  import gradio as gr
2
+ from google import genai
3
  import edge_tts
4
  import os
5
 
 
53
 
54
  async def band_consulting(user_input, member_name):
55
  try:
 
56
  if member_name not in MEMBERS:
57
  return f"Error: ์ด๋ฆ„ ๋ถˆ์ผ์น˜! ('{member_name}')", None
58
 
59
  member = MEMBERS[member_name]
60
 
61
+ # [Step 1] ๋‡Œ: ๊ฐ€์žฅ ์•ˆ์ •์ ์ธ 1.5 Flash ๋ชจ๋ธ ์‚ฌ์šฉ
62
  chat_prompt = f"{member['prompt']}\n\nUser's Error/Worry: {user_input}\nResponse:"
63
 
64
  try:
65
+ # โ˜… ๋ชจ๋ธ๋ช… ๋ณ€๊ฒฝ ์™„๋ฃŒ: gemini-1.5-flash
66
  response = client.models.generate_content(
67
+ model='gemini-1.5-flash',
68
  contents=chat_prompt
69
  )
70
  ai_text = response.text
71
  except Exception as e:
72
+ # ์—๋Ÿฌ ๋‚ด์šฉ์„ ๋” ์ž์„ธํžˆ ์ถœ๋ ฅ
73
  return f"Error (Gemini): {str(e)}", None
74
 
75
  # [Step 2] ์ž…: Edge-TTS
 
95
  out_audio = gr.Audio(label="์Œ์„ฑ")
96
  btn = gr.Button("์ƒ๋‹ด ์š”์ฒญ")
97
 
 
98
  btn.click(fn=band_consulting, inputs=[inp_text, inp_member], outputs=[out_text, out_audio], api_name="predict")
99
 
100
  demo.launch()