Update app.py

app.py

@@ -1,91 +1,91 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory
-import base64
-import re
-import os
-from datetime import datetime
-import requests
-
-# OpenAI API Key
-api_key = "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"
-
-app = Flask(__name__)
-
-# Function to encode the image
-def encode_image(image_path):
-  with open(image_path, "rb") as image_file:
-    return base64.b64encode(image_file.read()).decode('utf-8')
-
-@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)
-    
-    # Create a unique file name
-    timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
-    file_path = f'captured_image_{timestamp}.png'
-    
-    # Save the image to a file
-    with open(file_path, 'wb') as f:
-        f.write(image_data)
-    
-    
-
-    # Path to your image
-    image_path = file_path
-
-    # Getting the base64 string
-    base64_image = encode_image(image_path)
-
-    headers = {
-    "Content-Type": "application/json",
-    "Authorization": f"Bearer {api_key}"
-    }
-
-    payload = {
-    "model": "gpt-4o",
-    "messages": [
-        {
-        "role": "user",
-        "content": [
-            {
-            "type": "text",
-            "text": "이미지를 입력받으면 당류가 몇 g인지 예시와 같은 형식만 출력하시오.\n예) 당류 : 10g \n상품분석표가 아니라면 'error'를 출력하시오."
-            },
-            {
-            "type": "image_url",
-            "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_image}"
-            }
-            }
-        ]
-        }
-    ],
-    "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': '분석이 완료되었습니다.', 'image_url': file_path, 'analysis_result': analysis_result})
-
-
-
-@app.route('/images/<filename>')
-def get_image(filename):
-    return send_from_directory('.', filename)
-
-if __name__ == '__main__':
+from flask import Flask, request, jsonify, render_template, send_from_directory
+import base64
+import re
+import os
+from datetime import datetime
+import requests
+
+# OpenAI API Key
+api_key = "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"
+
+app = Flask(__name__)
+
+# Function to encode the image
+def encode_image(image_path):
+  with open(image_path, "rb") as image_file:
+    return base64.b64encode(image_file.read()).decode('utf-8')
+
+@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)
+    
+    # Create a unique file name
+    timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
+    file_path = os.path.join('uploads', f'captured_image_{timestamp}.png')
+    
+    # Save the image to a file
+    with open(file_path, 'wb') as f:
+        f.write(image_data)
+    
+    
+
+    # Path to your image
+    image_path = file_path
+
+    # Getting the base64 string
+    base64_image = encode_image(image_path)
+
+    headers = {
+    "Content-Type": "application/json",
+    "Authorization": f"Bearer {api_key}"
+    }
+
+    payload = {
+    "model": "gpt-4o",
+    "messages": [
+        {
+        "role": "user",
+        "content": [
+            {
+            "type": "text",
+            "text": "이미지를 입력받으면 당류가 몇 g인지 예시와 같은 형식만 출력하시오.\n예) 당류 : 10g \n상품분석표가 아니라면 'error'를 출력하시오."
+            },
+            {
+            "type": "image_url",
+            "image_url": {
+                "url": f"data:image/jpeg;base64,{base64_image}"
+            }
+            }
+        ]
+        }
+    ],
+    "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': '분석이 완료되었습니다.', 'image_url': file_path, 'analysis_result': analysis_result})
+
+
+
+@app.route('/images/<filename>')
+def get_image(filename):
+    return send_from_directory('.', filename)
+
+if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)