devlim commited on
Commit
5b7691d
ยท
verified ยท
1 Parent(s): 1c96f1e

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -89
app.py DELETED
@@ -1,89 +0,0 @@
1
- from flask import Flask, request, jsonify, render_template
2
- import base64
3
- import re
4
- import requests
5
- import os
6
-
7
- app = Flask(__name__)
8
-
9
- # Get API keys from environment variables
10
- api_key = os.getenv('OPENAI_API_KEY')
11
- huggingface_api_key = os.getenv('HUGGINGFACE_API_KEY')
12
- huggingface_url = "https://huggingface.co/api/spaces/devlim/supernova/upload"
13
-
14
- @app.route('/')
15
- def index():
16
- return render_template('index.html')
17
-
18
- @app.route('/save_image', methods=['POST'])
19
- def save_image():
20
- if 'image' not in request.files:
21
- return jsonify({'message': 'No image part in the request'}), 400
22
-
23
- file = request.files['image']
24
-
25
- if file.filename == '':
26
- return jsonify({'message': 'No selected file'}), 400
27
-
28
- # Save the image to a temporary file
29
- temp_image_path = os.path.join("temp", file.filename)
30
- file.save(temp_image_path)
31
-
32
- # Upload the image to Hugging Face Space
33
- headers = {
34
- "Authorization": f"Bearer {huggingface_api_key}"
35
- }
36
- files = {
37
- 'file': (file.filename, open(temp_image_path, 'rb'), 'image/png')
38
- }
39
- response = requests.post(huggingface_url, headers=headers, files=files)
40
-
41
- if response.status_code != 200:
42
- os.remove(temp_image_path)
43
- return jsonify({'message': 'Error: ํ—ˆ๊น…ํŽ˜์ด์Šค์— ์ด๋ฏธ์ง€๋ฅผ ์—…๋กœ๋“œํ•  ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค.'}), 500
44
-
45
- # Get the uploaded image URL from Hugging Face response
46
- uploaded_image_url = response.json()['url']
47
-
48
- # Prepare the payload for OpenAI API
49
- headers = {
50
- "Content-Type": "application/json",
51
- "Authorization": f"Bearer {api_key}"
52
- }
53
- payload = {
54
- "model": "gpt-4",
55
- "messages": [
56
- {
57
- "role": "user",
58
- "content": [
59
- {
60
- "type": "text",
61
- "text": "์ด๋ฏธ์ง€๋ฅผ ์ž…๋ ฅ๋ฐ›์œผ๋ฉด ๋‹น๋ฅ˜๊ฐ€ ๋ช‡ g์ธ์ง€ ์˜ˆ์‹œ์™€ ๊ฐ™์€ ํ˜•์‹๋งŒ ์ถœ๋ ฅํ•˜์‹œ์˜ค.\n์˜ˆ) ๋‹น๋ฅ˜ : 10g \n์ƒํ’ˆ๋ถ„์„ํ‘œ๊ฐ€ ์•„๋‹ˆ๋ผ๋ฉด 'error'๋ฅผ ์ถœ๋ ฅํ•˜์‹œ์˜ค."
62
- },
63
- {
64
- "type": "image_url",
65
- "image_url": {
66
- "url": uploaded_image_url
67
- }
68
- }
69
- ]
70
- }
71
- ],
72
- "max_tokens": 300
73
- }
74
-
75
- response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
76
-
77
- os.remove(temp_image_path)
78
-
79
- if response.status_code == 200:
80
- result = response.json()
81
- analysis_result = result['choices'][0]['message']['content']
82
- else:
83
- analysis_result = "Error: ๋‹น๋ฅ˜๋ฅผ ์ฐพ์„ ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค."
84
-
85
- return jsonify({'message': '๋ถ„์„์ด ์™„๋ฃŒ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.', 'analysis_result': analysis_result})
86
-
87
- if __name__ == '__main__':
88
- os.makedirs('temp', exist_ok=True)
89
- app.run(host='0.0.0.0', port=7860, debug=True)