Update app.py

app.py

@@ -1,67 +1,63 @@
+import os, json, requests
 from flask import Flask, request
-from gradio_client import Client
-from uuid import uuid4
-import os, json, requests, hashlib
 
 app = Flask(__name__)
 
-secret = os.environ.get("auth")
-headers = {'Authorization': os.environ.get("auth")}
-
-res_dict = {}
-
 @app.route('/', methods=['GET'])
 def index():
     return {}, 200
 
-@app.route('/queue', methods=['POST'])
-def queue():
-    if request.json.get('secret') != secret:
-        return {}, 400
-        
-    uuid = str(uuid4())
-    res_dict[uuid] = request.json
-    
-    return { 'uuid': uuid }, 200
-
-
-@app.route('/6asr', methods=['POST'])
-def sixasr():
-    data = res_dict.get(request.json.get('uuid'))
-    if not data:
-        return {}, 400
-
-    base64 = request.json.get('base64')
-    dialectCode = request.json.get('dialectCode')
-    if not base64 or not dialectCode:
-        return {}, 400
-    
-    if hashlib.sha256(base64.encode('utf-8')).hexdigest() != data.get("base64Hash"):
-        return {}, 400
-
-    URL = 'https://nuu-lab-gradio-6asr-whisper.hf.space/run/predict'
-    body = {'data': ["whisper-large-v3", f"htia_{dialectCode}", None, base64, '']}
-
-    res = requests.post(URL, json=body, headers=headers)
-    if res.status_code == 200:
-        res_json = res.json()
-
-        callbackURL = data.get('callbackURL')
-        callback_body = data.get('callbackBody')
-
-        if res_json.get('data'):
-            callback_body['length'] = len(res_json['data'][0])
-        if res_json.get('srt'):
-            callback_body['srt'] = res_json['srt'][0]
-
-        requests.post(callbackURL, json=callback_body)
-
-        print(callback_body)
-        print(res_json)
-
-        return res_json
-
-    return {'error': '請求失敗'}, 500
+LINE_REPLY_URL = "https://api.line.me/v2/bot/message/reply"
+LINE_CHANNEL_ACCESS_TOKEN = os.environ.get("LINE_CHANNEL_ACCESS_TOKEN")
+
+LINE_REPLY_HEADERS = {
+    "Content-Type": "application/json; charset=UTF-8",
+    "Authorization": "Bearer " + LINE_CHANNEL_ACCESS_TOKEN
+}
+
+@app.route("/api/", methods=["POST"])
+def api():
+    try:
+        payload = get_payload_dict(request.get_json())
+        print(payload)
+        # res_text = opai.private_translate_api(payload.get("msg_text"))
+        # send_text(payload.get("token"), res_text)
+    except:
+        pass
+    return "", 200
+
+
+def get_payload_dict(raw_payload):
+    print(raw_payload)
+    token = raw_payload.get("events", [{}])[0].get("replyToken")
+    sender_type = raw_payload.get("events", [{}])[0].get("source", {}).get("type", "user")
+    sender_id = raw_payload.get("events", [{}])[0].get("source", {}).get(str(sender_type+"Id"))
+    msg_type = raw_payload.get("events", [{}])[0].get("message", {}).get("type")
+    msg_id = raw_payload.get("events", [{}])[0].get("message", {}).get("id")
+    msg_text = raw_payload.get("events", [{}])[0].get("message", {}).get("text")
+    postback_data = raw_payload.get("events", [{}])[0].get("postback", {}).get("data")
+
+    payload = {"token": token,
+               "sender_type": sender_type,
+               "sender_id": sender_id,
+               "msg_type": msg_type,
+               "msg_id": msg_id,
+               "msg_text": msg_text,
+               "postback_data": postback_data
+               }
+    return payload
+
+
+def send_text(token, text):
+    res = requests.post(LINE_REPLY_URL, headers=LINE_REPLY_HEADERS, json={
+        "replyToken": token,
+        "messages": [
+            {
+                "type": "text",
+                "text": text
+            }
+        ]
+    })
 
 
 if __name__ == "__main__":