Rooni commited on
Commit
0561887
·
verified ·
1 Parent(s): 594642f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -11
app.py CHANGED
@@ -7,6 +7,7 @@ from langdetect import detect
7
  from gradio_client import Client, handle_file
8
  import gradio as gr
9
  from PIL import Image
 
10
 
11
  # Функция для получения случайного API ключа
12
  def get_random_api_key():
@@ -71,26 +72,40 @@ def process_image_and_text(image, text):
71
  print(f"Ошибка при обработке: {e}")
72
  return None
73
 
 
 
 
 
 
 
 
 
 
 
 
74
  # Функция для получения примеров
75
  def get_samples():
76
  sample_list = [
77
  {
78
- "image": "img/example_1in.png",
79
  "text": "Луна, данный предмет лежит на поверхности луны — груша. На заднем плане Земля возвышается на переднем плане, ультра реализм",
80
  },
 
 
 
 
 
 
 
 
81
  ]
82
 
83
  samples_with_images = []
84
  for sample in sample_list:
85
- image_path = sample["image"]
86
- if os.path.exists(image_path):
87
- try:
88
- img = Image.open(image_path)
89
- samples_with_images.append([img, sample["text"]])
90
- except Exception as e:
91
- print(f"Ошибка при открытии изображения {image_path}: {e}")
92
- else:
93
- print(f"Файл изображения {image_path} не найден.")
94
 
95
  return samples_with_images
96
 
@@ -122,4 +137,4 @@ if __name__ == "__main__":
122
  demo.launch(
123
  show_api=False,
124
  share=False
125
- )
 
7
  from gradio_client import Client, handle_file
8
  import gradio as gr
9
  from PIL import Image
10
+ from io import BytesIO
11
 
12
  # Функция для получения случайного API ключа
13
  def get_random_api_key():
 
72
  print(f"Ошибка при обработке: {e}")
73
  return None
74
 
75
+ # Функция для загрузки изображения из URL
76
+ def load_image_from_url(url):
77
+ try:
78
+ response = requests.get(url, timeout=10)
79
+ response.raise_for_status()
80
+ image = Image.open(BytesIO(response.content))
81
+ return image
82
+ except Exception as e:
83
+ print(f"Ошибка при загрузке изображения из URL {url}: {e}")
84
+ return None
85
+
86
  # Функция для получения примеров
87
  def get_samples():
88
  sample_list = [
89
  {
90
+ "image": "https://example.com/example_1in.png", # Замените на реальный URL
91
  "text": "Луна, данный предмет лежит на поверхности луны — груша. На заднем плане Земля возвышается на переднем плане, ультра реализм",
92
  },
93
+ {
94
+ "image": "https://example.com/example_2in.png", # Замените на реальный URL
95
+ "text": "Пример 2. Описание изображения 2",
96
+ },
97
+ {
98
+ "image": "https://example.com/example_3in.png", # Замените на реальный URL
99
+ "text": "Пример 3. Описание изображения 3. Более длинное описание, чтобы продемонстрировать разный размер текста.",
100
+ }
101
  ]
102
 
103
  samples_with_images = []
104
  for sample in sample_list:
105
+ image_url = sample["image"]
106
+ image = load_image_from_url(image_url)
107
+ if image:
108
+ samples_with_images.append([image, sample["text"]])
 
 
 
 
 
109
 
110
  return samples_with_images
111
 
 
137
  demo.launch(
138
  show_api=False,
139
  share=False
140
+ )