AI/app.py at main · Petlendar/AI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
from flask import Flask, request, jsonify
from flask_cors import CORS
import openai
from FoodRec import process_food_rec
from PetAdvice import process_pet_advice
from PetMonitor import process_pet_monitor
from VaccineInfo import process_vaccine_info
from GetData import pet_info, vaccine_info
from GetText import extract_text


import os
from dotenv import load_dotenv  # .env 파일에서 환경 변수 로드

app = Flask(__name__)

CORS(app,
     resources={r"/*": {"origins": "http://localhost:5001"}},
     methods=["GET", "POST", "OPTIONS"],
     allow_headers=["Content-Type", "Authorization"],
     supports_credentials=True)

load_dotenv()

api_key = os.getenv("OPENAI_API_KEY")

@app.route('/api/ai/food', methods=['POST'])
def food_recommendation():
    try:
        data = request.get_json()
        print("Received data:", data)
        extracted_data = extract_text(data)
        if isinstance(extracted_data, tuple):
            text, jwt_token, pet_id = extracted_data
        else:
            return extracted_data # 예외 처리

        result = process_food_rec(text, api_key, pet_id, jwt_token)
        return jsonify({"result": result})

    except Exception as e:
        return jsonify({"result": f"An unexpected error occurred: {str(e)}"}), 500

@app.route('/api/ai/petadvice', methods=['POST'])
def pet_advisor():
    try:
        data = request.get_json()
        print("Received data:", data)
        extracted_data = extract_text(data)
        if isinstance(extracted_data, tuple):
            text, jwt_token, pet_id = extracted_data
        else:
            return extracted_data

        result = process_pet_advice(text, api_key, pet_id, jwt_token)
        return jsonify({"result": result})

    except Exception as e:
        return jsonify({"result": f"An unexpected error occurred: {str(e)}"}), 500

@app.route('/api/ai/petmonitor', methods=['POST'])
def pet_monotoring():
    try:
        data = request.get_json()
        print("Received data:", data)
        extracted_data = extract_text(data)
        if isinstance(extracted_data, tuple):
            text, jwt_token, pet_id = extracted_data
        else:
            return extracted_data

        result = process_pet_monitor(text, api_key, pet_id, jwt_token)
        return jsonify({"result": result})

    except Exception as e:
        return jsonify({"result": f"An unexpected error occurred: {str(e)}"}), 500

@app.route('/api/ai/vaccine', methods=['POST'])
def vaccine_information():
    try:
        data = request.get_json()
        print("Received data:", data)
        extracted_data = extract_text(data)
        if isinstance(extracted_data, tuple):
            text, jwt_token, pet_id = extracted_data
        else:
            return extracted_data

        result = process_vaccine_info(text, api_key, pet_id, jwt_token)
        return jsonify({"result": result})

    except Exception as e:
        return jsonify({"result": f"An unexpected error occurred: {str(e)}"}), 500

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)