Create app.py

app.py

@@ -0,0 +1,90 @@
+from flask import Flask, request, jsonify, render_template
+import base64
+import re
+import requests
+
+# OpenAI API Key
+api_key_1 = "sk-"
+api_key_2 = "Ts4M29N6u2rPPzsrCy2qT3BlbkFJu1z6otKVXaJAbaIvIesj"
+huggingface_api_key_1 = "hf_"
+huggingface_api_key_2 = "RIYBqKlSJQSvLeQuWuTXuohTuhzVMMWBZR"
+
+api_key = api_key_1 + api_key_2
+huggingface_api_key = huggingface_api_key_1 + huggingface_api_key_2
+
+huggingface_url = "https://huggingface.co/spaces/devlim/supernova/upload"
+
+app = Flask(__name__)
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/save_image', methods=['POST'])
+def save_image():
+    data = request.get_json()
+    image_data = data['image']
+    
+    # Decode the base64 image data
+    image_data = re.sub('^data:image/.+;base64,', '', image_data)
+    image_data = base64.b64decode(image_data)
+    
+    # Save image to temporary file
+    temp_filename = "temp_image.png"
+    with open(temp_filename, "wb") as temp_file:
+        temp_file.write(image_data)
+    
+    # Upload image to Hugging Face
+    headers = {
+        "Authorization": f"Bearer {huggingface_api_key}"
+    }
+    files = {
+        'file': (temp_filename, open(temp_filename, 'rb'), 'image/png')
+    }
+    response = requests.post(huggingface_url, headers=headers, files=files)
+    
+    if response.status_code != 200:
+        return jsonify({'message': 'Error: 허깅페이스에 이미지를 업로드할 수 없습니다.'}), 500
+    
+    # Get uploaded image URL
+    uploaded_image_url = response.json()['url']
+    
+    # Prepare payload for OpenAI API
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+    payload = {
+        "model": "gpt-4",
+        "messages": [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": "이미지를 입력받으면 당류가 몇 g인지 예시와 같은 형식만 출력하시오.\n예) 당류 : 10g \n상품분석표가 아니라면 'error'를 출력하시오."
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": uploaded_image_url
+                        }
+                    }
+                ]
+            }
+        ],
+        "max_tokens": 300
+    }
+
+    response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
+
+    if response.status_code == 200:
+        result = response.json()
+        analysis_result = result['choices'][0]['message']['content']
+    else:
+        analysis_result = "Error: 당류를 찾을 수 없습니다."
+    
+    return jsonify({'message': '분석이 완료되었습니다.', 'analysis_result': analysis_result})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)