Update app.py

app.py

@@ -9,35 +9,37 @@ from prompts.main_prompt import MAIN_PROMPT
 if os.path.exists(".env"):
     load_dotenv(".env")
 
-OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
 
+
 def gpt_call(history, user_message,
              model="gpt-4o-mini",
              max_tokens=512,
              temperature=0.7,
              top_p=0.95):
     """
-    OpenAI ChatCompletion API function.
+    OpenAI ChatCompletion API를 통해 답변을 생성하는 함수.
+    - history: [(user_text, assistant_text), ...]
+    - user_message: 사용자가 방금 입력한 메시지
     """
-
-    # Ensure history is formatted correctly
+    # 1) 시스템 메시지(=MAIN_PROMPT)를 가장 앞에 추가
     messages = [{"role": "system", "content": MAIN_PROMPT}]
-
-    for exchange in history:
-        if isinstance(exchange, (list, tuple)) and len(exchange) == 2:
-            user_text, assistant_text = exchange
-            if isinstance(user_text, str) and user_text.strip():
-                messages.append({"role": "user", "content": user_text})
-            if isinstance(assistant_text, str) and assistant_text.strip():
-                messages.append({"role": "assistant", "content": assistant_text})
-
-    # Add latest user input
+    
+    # 2) 기존 대화 기록(history)을 OpenAI 형식으로 변환
+    #    user_text -> 'user' / assistant_text -> 'assistant'
+    for user_text, assistant_text in history:
+        if user_text:
+            messages.append({"role": "user", "content": user_text})
+        if assistant_text:
+            messages.append({"role": "assistant", "content": assistant_text})
+
+    # 3) 마지막에 이번 사용자의 입력을 추가
     messages.append({"role": "user", "content": user_message})
 
-    # Call OpenAI API
+    # 4) OpenAI API 호출
     completion = client.chat.completions.create(
         model=model,
         messages=messages,
@@ -45,7 +47,6 @@ def gpt_call(history, user_message,
         temperature=temperature,
         top_p=top_p
     )
-
     return completion.choices[0].message.content
 
 def respond(user_message, history):
@@ -85,7 +86,7 @@ with gr.Blocks() as demo:
 
     # (user, assistant) 쌍을 저장할 히스토리 상태
     # 여기서도 동일한 초기 상태를 넣어줌
-    state_history = gr.State([[{"role": "assistant", "content": INITIAL_PROMPT}]])
+    state_history = gr.State([("", INITIAL_PROMPT)])
 
     # 사용자 입력
     user_input = gr.Textbox(