Update app.py

app.py

@@ -1,36 +1,55 @@
-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
-api_key = "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"
+import os
 
 app = Flask(__name__)
 
+# Get API keys from environment variables
+api_key = os.getenv('OPENAI_API_KEY')
+huggingface_api_key = os.getenv('HUGGINGFACE_API_KEY')
+huggingface_url = "https://huggingface.co/api/spaces/devlim/supernova/upload"
+
 @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']
+    if 'image' not in request.files:
+        return jsonify({'message': 'No image part in the request'}), 400
+
+    file = request.files['image']
     
-    # Decode the base64 image data
-    image_data = re.sub('^data:image/.+;base64,', '', image_data)
-    image_data = base64.b64decode(image_data)
+    if file.filename == '':
+        return jsonify({'message': 'No selected file'}), 400
+
+    # Save the image to a temporary file
+    temp_image_path = os.path.join("temp", file.filename)
+    file.save(temp_image_path)
+
+    # Upload the image to Hugging Face Space
+    headers = {
+        "Authorization": f"Bearer {huggingface_api_key}"
+    }
+    files = {
+        'file': (file.filename, open(temp_image_path, 'rb'), 'image/png')
+    }
+    response = requests.post(huggingface_url, headers=headers, files=files)
     
-    # Encode image to base64 string
-    base64_image = base64.b64encode(image_data).decode('utf-8')
+    if response.status_code != 200:
+        os.remove(temp_image_path)
+        return jsonify({'message': 'Error: 허깅페이스에 이미지를 업로드할 수 없습니다.'}), 500
 
+    # Get the uploaded image URL from Hugging Face response
+    uploaded_image_url = response.json()['url']
+
+    # Prepare the payload for OpenAI API
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {api_key}"
     }
-
     payload = {
         "model": "gpt-4",
         "messages": [
@@ -44,7 +63,7 @@ def save_image():
                     {
                         "type": "image_url",
                         "image_url": {
-                            "url": f"data:image/jpeg;base64,{base64_image}"
+                            "url": uploaded_image_url
                         }
                     }
                 ]
@@ -55,14 +74,16 @@ def save_image():
 
     response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
 
+    os.remove(temp_image_path)
+
     if response.status_code == 200:
         result = response.json()
         analysis_result = result['choices'][0]['message']['content']
     else:
         analysis_result = "Error: 당류를 찾을 수 없습니다."
 
-    print(analysis_result)
     return jsonify({'message': '분석이 완료되었습니다.', 'analysis_result': analysis_result})
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True)
+    os.makedirs('temp', exist_ok=True)
+    app.run(host='0.0.0.0', port=7860, debug=True)