Update app.py

app.py

@@ -10,10 +10,15 @@ api_key = "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"
 
 app = Flask(__name__)
 
+# Ensure the directory exists and set appropriate permissions
+uploads_dir = 'uploads'
+os.makedirs(uploads_dir, exist_ok=True)
+os.chmod(uploads_dir, 0o777)
+
 # 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')
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode('utf-8')
 
 @app.route('/')
 def index():
@@ -30,13 +35,11 @@ def save_image():
     
     # 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')
+    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
@@ -45,30 +48,30 @@ def save_image():
     base64_image = encode_image(image_path)
 
     headers = {
-    "Content-Type": "application/json",
-    "Authorization": f"Bearer {api_key}"
+        "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'를 출력하시오."
-            },
+        "model": "gpt-4",
+        "messages": [
             {
-            "type": "image_url",
-            "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_image}"
+                "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
+        ],
+        "max_tokens": 300
     }
 
     response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
@@ -81,8 +84,6 @@ def save_image():
 
     return jsonify({'message': '분석이 완료되었습니다.', 'image_url': file_path, 'analysis_result': analysis_result})
 
-
-
 @app.route('/images/<filename>')
 def get_image(filename):
     return send_from_directory('.', filename)