Update app.py

app.py

@@ -85,7 +85,7 @@ def respond(
 
     try:
         response = ""
-        for message in client.chat.completions.create(
+        for chunk in client.chat.completions.create(
             model="CohereForAI/c4ai-command-r-plus-08-2024",
             max_tokens=max_tokens,
             stream=True,
@@ -93,16 +93,16 @@ def respond(
             top_p=top_p,
             messages=messages,
         ):
-            if hasattr(message.choices[0].delta, 'content'):
-                token = message.choices[0].delta.content
+            if hasattr(chunk.choices[0].delta, 'content'):
+                token = chunk.choices[0].delta.content
                 if token is not None:
                     response += token
-                    yield history + [[message, response]]
+                    yield [(message, response)]
         
-        return history + [[message, response]]
+        return [(message, response)]
     except Exception as e:
         print(f"Error in respond: {str(e)}")
-        return history + [[message, f"오류가 발생했습니다: {str(e)}"]]
+        return [(message, f"오류가 발생했습니다: {str(e)}")]
 
 def process_chat(message, history):
     try:
@@ -111,11 +111,11 @@ def process_chat(message, history):
             params = extract_parameters(message, relevant_data)
             protein_result = generate_protein(params)
             explanation = generate_explanation(protein_result, params)
-            return history + [[message, explanation]]
+            return history + [(message, explanation)]
         else:
-            return history + [[message, "단백질 생성 관련 키워드를 포함해주세요."]]
+            return history + [(message, "단백질 생성 관련 키워드를 포함해주세요.")]
     except Exception as e:
-        return history + [[message, f"오류가 발생했습니다: {str(e)}"]]
+        return history + [(message, f"오류가 발생했습니다: {str(e)}")]
 
 def search_protein_data(query):
     relevant_entries = []
@@ -620,8 +620,11 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         with gr.Column(scale=1):
             # 챗봇 인터페이스
             gr.Markdown("# 🤖 AI 단백질 설계 도우미")
-            chatbot = gr.Chatbot(height=600)
-            # 여기에 수정된 입력 필드 추가 --------------------------------
+            # 여기를 수정
+            chatbot = gr.Chatbot(
+                height=600,
+                type='messages'  # 메시지 형식 지정
+            )
             with gr.Row():
                 msg = gr.Textbox(
                     label="메시지를 입력하세요",
@@ -632,6 +635,8 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                 submit_btn = gr.Button("전송", variant="primary", scale=1)
             clear = gr.Button("대화 내용 지우기")
 
+
+
             
             with gr.Accordion("채팅 설정", open=False):
                 system_message = gr.Textbox(