Update app.py

app.py

@@ -1,26 +1,13 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory
+from flask import Flask, request, jsonify, render_template
 import base64
 import re
-import os
-from datetime import datetime
 import requests
 
-# OpenAI API Key
+app = Flask(__name__)
 
 # OpenAI API Key
 api_key = "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"
 
-app = Flask(__name__)
-
-# Ensure the directory exists
-uploads_dir = 'uploads'
-os.makedirs(uploads_dir, exist_ok=True)
-
-# 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')
@@ -32,21 +19,7 @@ def save_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_dir, 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)
+    base64_image = image_data  # This is the base64 string of the image
 
     headers = {
         "Content-Type": "application/json",
@@ -58,18 +31,14 @@ def save_image():
         "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}"
-                        }
-                    }
-                ]
+                "content": "이미지를 입력받으면 당류가 몇 g인지 예시와 같은 형식만 출력하시오.\n예) 당류 : 10g \n상품분석표가 아니라면 'error'를 출력하시오."
+            },
+            {
+                "role": "user",
+                "content": {
+                    "type": "image",
+                    "image": f"data:image/png;base64,{base64_image}"
+                }
             }
         ],
         "max_tokens": 300
@@ -83,11 +52,7 @@ def save_image():
     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)
+    return jsonify({'message': '분석이 완료되었습니다.', 'analysis_result': analysis_result})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)