Update app.py

app.py

@@ -68,7 +68,7 @@ ds = load_dataset("lamm-mit/protein_secondary_structure_from_PDB",
 
 def respond(
     message,
-    history: list[tuple[str, str]],
+    history,
     system_message,
     max_tokens,
     temperature,
@@ -76,10 +76,10 @@ def respond(
 ):
     messages = [{"role": "system", "content": system_message}]
 
-    for user_msg, assistant_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
+    for msg in history:
+        messages.append({"role": "user", "content": msg[0]})
+        if msg[1]:
+            messages.append({"role": "assistant", "content": msg[1]})
 
     messages.append({"role": "user", "content": message})
 
@@ -97,12 +97,15 @@ def respond(
                 token = chunk.choices[0].delta.content
                 if token is not None:
                     response += token
-                    yield [(message, response)]
+                    yield [{"role": "user", "content": message},
+                          {"role": "assistant", "content": response}]
         
-        return [(message, response)]
+        return [{"role": "user", "content": message},
+                {"role": "assistant", "content": response}]
     except Exception as e:
         print(f"Error in respond: {str(e)}")
-        return [(message, f"오류가 발생했습니다: {str(e)}")]
+        return [{"role": "user", "content": message},
+                {"role": "assistant", "content": f"오류가 발생했습니다: {str(e)}"}]
 
 def process_chat(message, history):
     try:
@@ -111,11 +114,20 @@ 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 + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": explanation}
+            ]
         else:
-            return history + [(message, "단백질 생성 관련 키워드를 포함해주세요.")]
+            return history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": "단백질 생성 관련 키워드를 포함해주세요."}
+            ]
     except Exception as e:
-        return history + [(message, f"오류가 발생했습니다: {str(e)}")]
+        return history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": f"오류가 발생했습니다: {str(e)}"}
+        ]
 
 def search_protein_data(query):
     relevant_entries = []
@@ -624,6 +636,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             chatbot = gr.Chatbot(
                 height=600,
                 type='messages'  # 메시지 형식 지정
+          
             )
             with gr.Row():
                 msg = gr.Textbox(