Update app.py

app.py

@@ -13,8 +13,6 @@ from supabase import create_client, Client
 from datetime import datetime
 import requests
 import logging
-from google.cloud import translate_v2 as translate
-import google.generativeai as genai
 
 # Configuração de logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -29,17 +27,8 @@ supabase: Client = create_client(url, key)
 hf_token = os.getenv("HF_TOKEN")
 
 # Configura as chaves de API da OpenAI e Google AI
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "Openai")
-GOOGLE_AI_KEY = os.getenv("GOOGLE_AI_KEY", "GoogleAi")
-
-# Configura o Google AI Gemini
-try:
-    genai.configure(api_key=GOOGLE_AI_KEY)
-    google_ai_available = True
-    logger.info("✅ Google AI Gemini configurado com sucesso")
-except Exception as e:
-    google_ai_available = False
-    logger.error(f"❌ Erro ao configurar Google AI Gemini: {e}")
+OPENAI_API_KEY = os.getenv("Openai", "")
+GOOGLE_AI_KEY = os.getenv("GoogleAi", "")
 
 # Inicializa o modelo base FLUX.1-dev
 base_model = "black-forest-labs/FLUX.1-dev"
@@ -158,7 +147,7 @@ def optimize_with_openai(text, character_option):
 
 def optimize_with_google_ai(text, character_option):
     """
-    Otimiza o prompt usando o Google AI Gemini.
+    Otimiza o prompt usando o Google AI API diretamente.
     """
     try:
         # Determina o prompt com base na seleção de personagem
@@ -181,28 +170,40 @@ def optimize_with_google_ai(text, character_option):
             Seu prompt deve ser detalhado, claro e deve enfatizar a presença dos DOIS personagens na mesma cena.
             Responda APENAS com o prompt otimizado em inglês, sem explicações ou introduções."""
         
-        # Configura o modelo Gemini
-        generation_config = {
-            "temperature": 0.7,
-            "top_p": 1,
-            "top_k": 32,
-            "max_output_tokens": 300,
+        # Prepara a requisição para a API do Google Gemini
+        headers = {
+            "Content-Type": "application/json"
         }
         
-        model = genai.GenerativeModel(
-            model_name="gemini-pro",
-            generation_config=generation_config
-        )
+        url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GOOGLE_AI_KEY}"
+        
+        data = {
+            "contents": [
+                {
+                    "parts": [
+                        {
+                            "text": f"{instruction}\n\nPrompt original: {text}"
+                        }
+                    ]
+                }
+            ],
+            "generationConfig": {
+                "temperature": 0.7,
+                "topK": 32,
+                "topP": 1,
+                "maxOutputTokens": 300
+            }
+        }
         
-        # Cria a conversa
-        convo = model.start_chat(history=[])
-        convo.send_message(f"{instruction}\n\nPrompt original: {text}")
+        # Faz a requisição para o Google AI
+        response = requests.post(url, headers=headers, json=data)
+        response.raise_for_status()
         
-        # Obtém a resposta
-        response = convo.last.text
-        logger.info(f"✅ Prompt otimizado via Google AI: {response}")
+        # Extrai a resposta
+        text_response = response.json()["candidates"][0]["content"]["parts"][0]["text"]
+        logger.info(f"✅ Prompt otimizado via Google AI: {text_response}")
         
-        return response.strip()
+        return text_response.strip()
     
     except Exception as e:
         logger.error(f"❌ Erro ao otimizar prompt com Google AI: {e}")
@@ -227,11 +228,7 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
     if ai_option == "OpenAI":
         processed_prompt = optimize_with_openai(prompt, character_option)
     elif ai_option == "Google AI":
-        if google_ai_available:
-            processed_prompt = optimize_with_google_ai(prompt, character_option)
-        else:
-            logger.warning("Google AI indisponível, usando OpenAI como fallback")
-            processed_prompt = optimize_with_openai(prompt, character_option)
+        processed_prompt = optimize_with_google_ai(prompt, character_option)
     elif ai_option == "Nenhum":
         # Mantém o prompt original, sem processamento
         processed_prompt = prompt
@@ -354,7 +351,7 @@ with gr.Blocks(theme=gr_theme) as app:
                 ai_option = gr.Radio(
                     ["Nenhum", "OpenAI", "Google AI"], 
                     label="Método de Otimização", 
-                    value="Google AI" if google_ai_available else "OpenAI"
+                    value="Google AI"
                 )
 
         with gr.Column(scale=2):
@@ -383,4 +380,4 @@ with gr.Blocks(theme=gr_theme) as app:
     )
 
 app.queue()
-app.launch(share=True)
+app.launch(share=True)