Update app.py
Browse files
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
|
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 |
-
|
43 |
-
|
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("##
|
76 |
|
77 |
-
|
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)
|