Update app.py

app.py

@@ -36,35 +36,72 @@ def fetch_message(df, num, direction):
 def create_prompt(instruction, num, tone, limit):
     return f"{instruction}\n## 制約条件\n- {num}文字以上\n- {tone}\n- {limit}\n"
 
+
 def respond(message, chat_history, system_prompt, model, api_key, temperature, max_tokens):
     try:
         client = anthropic.Anthropic(api_key=api_key)
-        messages = (
-            {"role": "system", "content": [{"type": "text", "text": system_prompt}]},
-            *[
-                {"role": "user", "content": [{"type": "text", "text": item[0]}]},
-                {"role": "assistant", "content": [{"type": "text", "text": item[1]}]}
-            ]
-            for item in chat_history
-        )
-        messages.append({"role": "user", "content": [{"type": "text", "text": message}]})
+        messages = []
+        for item in chat_history:
+            if isinstance(item, tuple):
+                usermessage = item[0]
+                assistantmessage = item[1]
+            elif isinstance(item, list):
+                usermessage = item[0]
+                assistantmessage = item[1]
+                if isinstance(usermessage, str) and usermessage.startswith("[('"):
+                    usermessage = eval(usermessage)[0][1]
+                    assistantmessage = eval(assistantmessage)[0][1]
+            else:
+                continue
+
+            messages.append({
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": usermessage
+                    }
+                ]
+            })
+            messages.append({
+                "role": "assistant",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": assistantmessage
+                    }
+                ]
+            })
 
+        messages.append(
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": message
+                    }
+                ]
+            }
+        )
         bot_message = ""
-        chat_history.append((message, bot_message))
+        chat_history.append((message,bot_message))
         with client.messages.stream(
             messages=messages,
+            system=system_prompt,
             model=model,
             max_tokens=max_tokens,
             temperature=temperature,
-        ) as stream:
-            for text in stream.text_stream:
-                bot_message += text
-                chat_history[-1] = (message, bot_message)
-                yield chat_history
+        )as stream:
+          for text in stream.text_stream:
+            bot_message += text
+            chat_history[-1]=(message, bot_message)
+            yield chat_history
     except Exception as e:
-        bot_message = str(e)
-        chat_history.append((message, bot_message))
-        return chat_history
+      bot_message = str(e)
+      chat_history.append((message, bot_message))
+      return chat_history
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# 生成AIのプロンプト術を学ぶクイックスタートガイド")