Update app.py

app.py

@@ -76,41 +76,46 @@ def respond(
 ):
     messages = [{"role": "system", "content": system_message}]
 
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    for user_msg, assistant_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
 
     messages.append({"role": "user", "content": message})
 
-    response = ""
-    
-    for message in client.chat.completions.create(
-        model="CohereForAI/c4ai-command-r-plus-08-2024",
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-        messages=messages,
-    ):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
-
-# 챗봇 및 단백질 생성 관련 함수들
+    try:
+        response = ""
+        for message in client.chat.completions.create(
+            model="CohereForAI/c4ai-command-r-plus-08-2024",
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+            messages=messages,
+        ):
+            if hasattr(message.choices[0].delta, 'content'):
+                token = message.choices[0].delta.content
+                if token is not None:
+                    response += token
+                    yield history + [[message, response]]
+        
+        return history + [[message, response]]
+    except Exception as e:
+        print(f"Error in respond: {str(e)}")
+        return history + [[message, f"오류가 발생했습니다: {str(e)}"]]
+
 def process_chat(message, history):
-    messages = [{"role": "user", "content": message}]
-    response = pipe(messages)[0]['generated_text']
-    
-    if any(keyword in message.lower() for keyword in ['protein', 'generate', '단백질', '생성']):
-        relevant_data = search_protein_data(message)
-        params = extract_parameters(response, relevant_data)
-        protein_result = generate_protein(params)
-        explanation = generate_explanation(protein_result, params)
-        return response + "\n\n" + explanation
-    
-    return response
+    try:
+        if any(keyword in message.lower() for keyword in ['protein', 'generate', '단백질', '생성']):
+            relevant_data = search_protein_data(message)
+            params = extract_parameters(message, relevant_data)
+            protein_result = generate_protein(params)
+            explanation = generate_explanation(protein_result, params)
+            return history + [[message, explanation]]
+        else:
+            return history + [[message, "단백질 생성 관련 키워드를 포함해주세요."]]
+    except Exception as e:
+        return history + [[message, f"오류가 발생했습니다: {str(e)}"]]
 
 def search_protein_data(query):
     relevant_entries = []