Update app.py

app.py

@@ -1,21 +1,51 @@
+from flask import Flask, request, jsonify
 import requests
 import os
 
+app = Flask(__name__)
+
 # Hugging Face APIエンドポイントと認証トークン
-api_url = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
-headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
-
-# クエリのメッセージ
-payload = {
-    "inputs": "Which of the depicted countries has the best food?",
-    "parameters": {"max_new_tokens": 512}
-}
-
-# APIリクエスト
-response = requests.post(api_url, headers=headers, json=payload)
-
-# 結果表示
-if response.status_code == 200:
-    print("Response:", response.json())
-else:
-    print(f"Error {response.status_code}: {response.text}")
+HF_API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+HEADERS = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+
+
+# /questionエンドポイント
+@app.route('/question', methods=['GET'])
+def question():
+    # パラメーターの取得
+    json_input = request.args.get("json")
+    text_input = request.args.get("text")
+
+    if not json_input and not text_input:
+        return jsonify({"error": "Please provide 'json' or 'text' parameter."}), 400
+
+    # 入力データの整形
+    if json_input:
+        try:
+            inputs = json_input
+        except Exception as e:
+            return jsonify({"error": f"Invalid JSON format: {str(e)}"}), 400
+    elif text_input:
+        inputs = text_input
+
+    # Hugging Face APIリクエストの送信
+    payload = {
+        "inputs": inputs,
+        "parameters": {"max_new_tokens": 512}
+    }
+
+    try:
+        response = requests.post(HF_API_URL, headers=HEADERS, json=payload)
+
+        if response.status_code == 200:
+            return jsonify({"response": response.json()})
+        else:
+            return jsonify({"error": f"Hugging Face API Error {response.status_code}: {response.text}"}), response.status_code
+
+    except Exception as e:
+        return jsonify({"error": f"Request failed: {str(e)}"}), 500
+
+
+if __name__ == '__main__':
+    # デバッグモードでサーバーを起動
+    app.run(debug=True, host="0.0.0.0", port=7860)