Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,35 @@ if os.path.exists(".env"):
|
|
12 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
13 |
client = OpenAI(api_key=OPENAI_API_KEY)
|
14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
# Reflection steps appear one-by-one
|
16 |
REFLECTION_STEPS = [
|
17 |
{
|
|
|
12 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
13 |
client = OpenAI(api_key=OPENAI_API_KEY)
|
14 |
|
15 |
+
# Define the function to call GPT model
|
16 |
+
def gpt_call(history, user_message, model="gpt-4o-mini", max_tokens=512, temperature=0.7, top_p=0.95):
|
17 |
+
"""
|
18 |
+
Calls OpenAI Chat API to generate responses.
|
19 |
+
- history: [(user_text, assistant_text), ...]
|
20 |
+
- user_message: latest message from user
|
21 |
+
"""
|
22 |
+
messages = [{"role": "system", "content": MAIN_PROMPT}]
|
23 |
+
|
24 |
+
# Add conversation history
|
25 |
+
for user_text, assistant_text in history:
|
26 |
+
if user_text:
|
27 |
+
messages.append({"role": "user", "content": user_text})
|
28 |
+
if assistant_text:
|
29 |
+
messages.append({"role": "assistant", "content": assistant_text})
|
30 |
+
|
31 |
+
messages.append({"role": "user", "content": user_message})
|
32 |
+
|
33 |
+
# OpenAI API Call
|
34 |
+
completion = client.chat.completions.create(
|
35 |
+
model=model,
|
36 |
+
messages=messages,
|
37 |
+
max_tokens=max_tokens,
|
38 |
+
temperature=temperature,
|
39 |
+
top_p=top_p
|
40 |
+
)
|
41 |
+
|
42 |
+
return completion.choices[0].message.content
|
43 |
+
|
44 |
# Reflection steps appear one-by-one
|
45 |
REFLECTION_STEPS = [
|
46 |
{
|