hmmmm

tools/image_generation_tools.py

@@ -88,8 +88,10 @@ class GenerateImageTool(Tool):
         """
         try:
             # Правильный импорт библиотеки
-            import google.generativeai as genai
-            from google.generativeai import types
+            from google import genai
+            from google.genai import types
+            from io import BytesIO
+            import base64
             
             # Get API key from environment variable
             api_key = os.environ.get("GEMINI_API_KEY")
@@ -98,8 +100,8 @@ class GenerateImageTool(Tool):
                 return "<p>Error: Gemini API key not found. Please set the GEMINI_API_KEY environment variable.</p>"
             
             logger.info("Initializing Gemini client")
-            genai.configure(api_key=api_key)
-            client = genai.Client()
+            # Новый способ настройки клиента
+            client = genai.Client(api_key=api_key)
             
             logger.info("Generating image with Gemini")
             response = client.models.generate_content(
@@ -114,10 +116,17 @@ class GenerateImageTool(Tool):
             for part in response.candidates[0].content.parts:
                 if part.text is not None:
                     logger.info(f"Gemini response text: {part.text[:100]}...")
-                elif hasattr(part, 'inline_data') and part.inline_data is not None:
-                    # Save the image to a temporary file
-                    # inline_data.data уже содержит данные в формате base64
-                    image_b64 = part.inline_data.data
+                elif part.inline_data is not None:
+                    # Извлекаем данные изображения
+                    image_data = part.inline_data.data
+                    
+                    # Преобразуем в base64 для HTML, если нужно
+                    if isinstance(image_data, bytes):
+                        image_b64 = base64.b64encode(image_data).decode('utf-8')
+                    else:
+                        # Если данные уже в base64
+                        image_b64 = image_data
+                        
                     img_html = f'<img src="data:image/png;base64,{image_b64}" alt="Generated image based on song analysis" style="max-width:100%; border-radius:10px; box-shadow:0 4px 8px rgba(0,0,0,0.1);">'
                     return img_html