alibicer commited on
Commit
b9e0046
ยท
verified ยท
1 Parent(s): 70aff7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -70
app.py CHANGED
@@ -5,105 +5,44 @@ from openai import OpenAI
5
  from prompts.initial_prompt import INITIAL_PROMPT
6
  from prompts.main_prompt import MAIN_PROMPT
7
 
8
- # .env ํŒŒ์ผ์—์„œ OPENAI_API_KEY ๋กœ๋“œ
9
  if os.path.exists(".env"):
10
  load_dotenv(".env")
11
 
12
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
13
-
14
  client = OpenAI(api_key=OPENAI_API_KEY)
15
 
16
-
17
-
18
- def gpt_call(history, user_message,
19
- model="gpt-4o-mini",
20
- max_tokens=512,
21
- temperature=0.7,
22
- top_p=0.95):
23
- """
24
- OpenAI ChatCompletion API๋ฅผ ํ†ตํ•ด ๋‹ต๋ณ€์„ ์ƒ์„ฑํ•˜๋Š” ํ•จ์ˆ˜.
25
- - history: [(user_text, assistant_text), ...]
26
- - user_message: ์‚ฌ์šฉ์ž๊ฐ€ ๋ฐฉ๊ธˆ ์ž…๋ ฅํ•œ ๋ฉ”์‹œ์ง€
27
- """
28
- # 1) ์‹œ์Šคํ…œ ๋ฉ”์‹œ์ง€(=MAIN_PROMPT)๋ฅผ ๊ฐ€์žฅ ์•ž์— ์ถ”๊ฐ€
29
  messages = [{"role": "system", "content": MAIN_PROMPT}]
30
 
31
- # 2) ๊ธฐ์กด ๋Œ€ํ™” ๊ธฐ๋ก(history)์„ OpenAI ํ˜•์‹์œผ๋กœ ๋ณ€ํ™˜
32
- # user_text -> 'user' / assistant_text -> 'assistant'
33
  for user_text, assistant_text in history:
34
  if user_text:
35
  messages.append({"role": "user", "content": user_text})
36
  if assistant_text:
37
  messages.append({"role": "assistant", "content": assistant_text})
38
 
39
- # 3) ๋งˆ์ง€๋ง‰์— ์ด๋ฒˆ ์‚ฌ์šฉ์ž์˜ ์ž…๋ ฅ์„ ์ถ”๊ฐ€
40
  messages.append({"role": "user", "content": user_message})
41
-
42
- # 4) OpenAI API ํ˜ธ์ถœ
43
- completion = client.chat.completions.create(
44
- model=model,
45
- messages=messages,
46
- max_tokens=max_tokens,
47
- temperature=temperature,
48
- top_p=top_p
49
- )
50
  return completion.choices[0].message.content
51
 
52
  def respond(user_message, history):
53
- """
54
- Gradio ์ƒ์—์„œ submitํ•  ๋•Œ ํ˜ธ์ถœ๋˜๋Š” ํ•จ์ˆ˜
55
- - user_message: ์‚ฌ์šฉ์ž๊ฐ€ ๋ฐฉ๊ธˆ ์นœ ๋ฉ”์‹œ์ง€
56
- - history: ๊ธฐ์กด (user, assistant) ํŠœํ”Œ ๋ฆฌ์ŠคํŠธ
57
- """
58
- # ์‚ฌ์šฉ์ž๊ฐ€ ๋นˆ ๋ฌธ์ž์—ด์„ ๋ณด๋ƒˆ๋‹ค๋ฉด ์•„๋ฌด ์ผ๋„ ํ•˜์ง€ ์•Š์Œ
59
  if not user_message:
60
  return "", history
61
 
62
- # GPT ๋ชจ๋ธ๋กœ๋ถ€ํ„ฐ ์‘๋‹ต์„ ๋ฐ›์Œ
63
  assistant_reply = gpt_call(history, user_message)
64
-
65
- # history์— (user, assistant) ์Œ ์ถ”๊ฐ€
66
  history.append((user_message, assistant_reply))
67
-
68
- # Gradio์—์„œ๋Š” (์ƒˆ๋กœ ๋น„์›Œ์งˆ ์ž…๋ ฅ์ฐฝ, ๊ฐฑ์‹ ๋œ history)๋ฅผ ๋ฐ˜ํ™˜
69
  return "", history
70
 
71
- ##############################
72
- # Gradio Blocks UI
73
- ##############################
74
  with gr.Blocks() as demo:
75
- gr.Markdown("## Simple Chat Interface")
76
 
77
- # Chatbot ์ดˆ๊ธฐ ์ƒํƒœ๋ฅผ ์„ค์ •
78
- # ์ฒซ ๋ฒˆ์งธ ๋ฉ”์‹œ์ง€๋Š” (user="", assistant=INITIAL_PROMPT) ํ˜•ํƒœ๋กœ ๋„ฃ์–ด
79
- # ํ™”๋ฉด์ƒ์—์„œ 'assistant'๊ฐ€ INITIAL_PROMPT๋ฅผ ๋งํ•œ ๊ฒƒ์ฒ˜๋Ÿผ ๋ณด์ด๊ฒŒ ํ•จ
80
- chatbot = gr.Chatbot(
81
- value=[("", INITIAL_PROMPT)], # (user, assistant)
82
- height=500
83
- )
84
-
85
- # (user, assistant) ์Œ์„ ์ €์žฅํ•  ํžˆ์Šคํ† ๋ฆฌ ์ƒํƒœ
86
- # ์—ฌ๊ธฐ์„œ๋„ ๋™์ผํ•œ ์ดˆ๊ธฐ ์ƒํƒœ๋ฅผ ๋„ฃ์–ด์คŒ
87
  state_history = gr.State([("", INITIAL_PROMPT)])
 
88
 
89
- # ์‚ฌ์šฉ์ž ์ž…๋ ฅ
90
- user_input = gr.Textbox(
91
- placeholder="Type your message here...",
92
- label="Your Input"
93
- )
94
-
95
- # ์ž…๋ ฅ์ด submit๋˜๋ฉด respond() ํ˜ธ์ถœ โ†’ ์ถœ๋ ฅ์€ (์ƒˆ ์ž…๋ ฅ์ฐฝ, ๊ฐฑ์‹ ๋œ chatbot)
96
- user_input.submit(
97
- respond,
98
- inputs=[user_input, state_history],
99
- outputs=[user_input, chatbot]
100
- ).then(
101
- # respond ๋๋‚œ ๋’ค, ์ตœ์‹  history๋ฅผ state_history์— ๋ฐ˜์˜
102
- fn=lambda _, h: h,
103
- inputs=[user_input, chatbot],
104
- outputs=[state_history]
105
- )
106
 
107
- # ๋ฉ”์ธ ์‹คํ–‰
108
  if __name__ == "__main__":
109
  demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
 
5
  from prompts.initial_prompt import INITIAL_PROMPT
6
  from prompts.main_prompt import MAIN_PROMPT
7
 
8
+ # Load OpenAI API Key from .env file
9
  if os.path.exists(".env"):
10
  load_dotenv(".env")
11
 
12
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
13
  client = OpenAI(api_key=OPENAI_API_KEY)
14
 
15
+ def gpt_call(history, user_message, model="gpt-4o-mini", max_tokens=1024, temperature=0.7, top_p=0.95):
 
 
 
 
 
 
 
 
 
 
 
 
16
  messages = [{"role": "system", "content": MAIN_PROMPT}]
17
 
 
 
18
  for user_text, assistant_text in history:
19
  if user_text:
20
  messages.append({"role": "user", "content": user_text})
21
  if assistant_text:
22
  messages.append({"role": "assistant", "content": assistant_text})
23
 
 
24
  messages.append({"role": "user", "content": user_message})
25
+
26
+ completion = client.chat.completions.create(model=model, messages=messages, max_tokens=max_tokens, temperature=temperature, top_p=top_p)
27
+
 
 
 
 
 
 
28
  return completion.choices[0].message.content
29
 
30
  def respond(user_message, history):
 
 
 
 
 
 
31
  if not user_message:
32
  return "", history
33
 
 
34
  assistant_reply = gpt_call(history, user_message)
 
 
35
  history.append((user_message, assistant_reply))
 
 
36
  return "", history
37
 
 
 
 
38
  with gr.Blocks() as demo:
39
+ gr.Markdown("## AI-Guided Math PD Chatbot")
40
 
41
+ chatbot = gr.Chatbot(value=[("", INITIAL_PROMPT)], height=600)
 
 
 
 
 
 
 
 
 
42
  state_history = gr.State([("", INITIAL_PROMPT)])
43
+ user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")
44
 
45
+ user_input.submit(respond, inputs=[user_input, state_history], outputs=[user_input, chatbot]).then(fn=lambda _, h: h, inputs=[user_input, chatbot], outputs=[state_history])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
 
 
47
  if __name__ == "__main__":
48
  demo.launch(server_name="0.0.0.0", server_port=7860, share=True)