Update app.py

app.py

@@ -15,19 +15,19 @@ MAX_INPUT_LENGTH = 1024
 INPUT_PROMPT = r'<s>\n以下は、タスクを説明する指示と、文脈のある入力の組み合わせです。要求を適切に満たす応答を書きなさい。\n[SEP]\n指示:\n{instruction}\n[SEP]\n入力:\n{input}\n[SEP]\n応答:\n'
 NO_INPUT_PROMPT = r'<s>\n以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。\n[SEP]\n指示:\n{instruction}\n[SEP]\n応答:\n'
 
-# デフォルトのロールインストラクション
-role_instruction = [
-    "User:睡眠に悩む高校生です",
-    "Assistant:では、お手伝いしましょう。！"
-]
-
-# デフォルトのチャットヒストリー
-default_conversation_history = [
-    "User: こんにちは、今日は一日を有効に使いたいのですが、何かアドバイスはありますか？",
-    "Assistant: こんにちは！一日の計画を立てることはとても重要です。朝、昼、晩それぞれの時間帯でやるべきことをリストにまとめると良いですよ。",
-    "User: なるほど、具体的にはどんな内容をリストに入れればいいですか？",
-    "Assistant: 朝は、起床後のルーチンや朝食の準備、健康的な運動を入れると良いですね。昼は、仕事や勉強の計画、休憩時間、昼食の準備が考えられます。晩は、夕食の準備や家事、リラックスタイム、就寝前のルーチンなどが含まれます。"
-]
+def get_default_role_instruction():
+    return [
+        "User:睡眠に悩む高校生です",
+        "Assistant:では、お手伝いしましょう。！"
+    ]
+
+def get_default_conversation_history():
+    return [
+        "User: こんにちは、今日は一日を有効に使いたいのですが、何かアドバイスはありますか？",
+        "Assistant: こんにちは！一日の計画を立てることはとても重要です。朝、昼、晩それぞれの時間帯でやるべきことをリストにまとめると良いですよ。",
+        "User: なるほど、具体的にはどんな内容をリストに入れればいいですか？",
+        "Assistant: 朝は、起床後のルーチンや朝食の準備、健康的な運動を入れると良いですね。昼は、仕事や勉強の計画、休憩時間、昼食の準備が考えられます。晩は、夕食の準備や家事、リラックスタイム、就寝前のルーチンなどが含まれます。"
+    ]
 
 def prepare_input(role_instruction, conversation_history, new_conversation):
     instruction = "".join([f"{text}\n" for text in role_instruction])
@@ -73,16 +73,21 @@ def generate_response(role_instruction, conversation_history, new_conversation):
 
 @app.route('/api', methods=['POST'])
 def api():
-    data = request.json
-    role_instruction = data.get('role_instruction', role_instruction)
-    conversation_history = data.get('conversation_history', default_conversation_history)
-    new_conversation = data.get('new_conversation', "")
+    try:
+        data = request.json
+        role_instruction = data.get('role_instruction', get_default_role_instruction())
+        conversation_history = data.get('conversation_history', get_default_conversation_history())
+        new_conversation = data.get('new_conversation', "")
 
-    if not new_conversation:
-        return jsonify({"error": "new_conversation is required"}), 400
+        if not new_conversation:
+            return jsonify({"error": "new_conversation is required"}), 400
 
-    formatted_output_all, response = generate_response(role_instruction, conversation_history, new_conversation)
-    return jsonify({"response": response, "conversation_history": conversation_history})
+        formatted_output_all, response = generate_response(role_instruction, conversation_history, new_conversation)
+        return jsonify({"response": response, "conversation_history": conversation_history})
+
+    except Exception as e:
+        print(f"Error: {e}")
+        return jsonify({"error": "An error occurred"}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)