Rooni commited on
Commit
c0cbb22
·
1 Parent(s): 6300ad8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -17
app.py CHANGED
@@ -14,27 +14,43 @@ def encode_image_to_base64(image):
14
 
15
  # Функция для отправки запроса в OpenAI с изображением и получения ответа
16
  def ask_openai_with_image(instruction, image):
17
- # Кодируем загруженное изображение в base64
18
- base64_image = encode_image_to_base64(image)
19
 
20
  # Убираем пробелы с начала и конца инструкции
21
  instruction = instruction.strip()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
 
23
- # Создаем данные для запроса с закодированным изображением
24
- payload = {
25
- "model": "gpt-4-vision-preview",
26
- "messages": [
27
- {
28
- "role": "user",
29
- "content": instruction,
30
- },
31
- {
32
- "role": "system",
33
- "content": f"data:image/jpeg;base64,{base64_image}",
34
- }
35
- ],
36
- "max_tokens": 4095,
37
- }
38
 
39
  # API ключ для OpenAI
40
  api_key = os.getenv("API_KEY")
 
14
 
15
  # Функция для отправки запроса в OpenAI с изображением и получения ответа
16
  def ask_openai_with_image(instruction, image):
 
 
17
 
18
  # Убираем пробелы с начала и конца инструкции
19
  instruction = instruction.strip()
20
+ if image != None
21
+ # Кодируем загруженное изображение в base64
22
+ base64_image = encode_image_to_base64(image)
23
+
24
+
25
+ # Создаем данные для запроса с закодированным изображением
26
+ payload = {
27
+ "model": "gpt-4-vision-preview",
28
+ "messages": [
29
+ {
30
+ "role": "user",
31
+ "content": instruction,
32
+ },
33
+ {
34
+ "role": "system",
35
+ "content": f"data:image/jpeg;base64,{base64_image}",
36
+ }
37
+ ],
38
+ "max_tokens": 4095,
39
+ }
40
+
41
+ if image == None
42
+ # Создаем данные для запроса с закодированным изображением
43
+ payload = {
44
+ "model": "gpt-4-vision-preview",
45
+ "messages": [
46
+ {
47
+ "role": "user",
48
+ "content": instruction,
49
+ }
50
+ ],
51
+ "max_tokens": 4095,
52
+ }
53
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
54
 
55
  # API ключ для OpenAI
56
  api_key = os.getenv("API_KEY")