Update app.py

app.py

@@ -300,5 +300,36 @@ with gr.Blocks(css=css, analytics_enabled=False) as demo:
     caption_cache = gr.State(value={"image_path": None, "caption": None})
 
     # обработчик клика по галерее: сразу подставляет картинку, генерирует подпись и обновляет кеш
-    def on_gallery_select(elem, cach
-    
+    def on_gallery_select(elem, cache):
+        # elem может быть строкой (URL) или список/tuple в некоторых версиях
+        img = None
+        if isinstance(elem, (list, tuple)):
+            img = elem[0] if elem else None
+        else:
+            img = elem
+        caption, new_cache = generate_and_cache_caption(img, cache)
+        # возвращаем (image_input value, raw_caption, new_cache)
+        return img, caption, new_cache
+
+    gallery.select(on_gallery_select, inputs=[gallery, caption_cache], outputs=[image_input, raw_caption, caption_cache])
+
+    # обработчик изменения image_input (загрузка своей картинки)
+    def on_image_change(image, cache):
+        caption, new_cache = generate_and_cache_caption(image, cache)
+        return caption, new_cache
+
+    image_input.change(on_image_change, inputs=[image_input, caption_cache], outputs=[raw_caption, caption_cache])
+
+    # отправка (кнопка или Enter) — стриминг LLM, outputs: (chatbot, raw_caption)
+    send_btn.click(chat_stream, inputs=[image_input, user_input, chatbot, caption_cache], outputs=[chatbot, raw_caption])
+    user_input.submit(chat_stream, inputs=[image_input, user_input, chatbot, caption_cache], outputs=[chatbot, raw_caption])
+
+    # очистка: сбрасываем чат и кеш
+    def clear_all():
+        return [], {"image_path": None, "caption": None}, ""
+
+    clear_btn.click(clear_all, inputs=None, outputs=[chatbot, caption_cache, raw_caption])
+
+# Запуск (в HF Spaces не указывать share=True)
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))