Spaces:

habulaj
/

newapi-clone

Running

App Files Files Community

habulaj commited on 1 day ago

Commit

75bb18b

verified ·

1 Parent(s): b06a19f

Update routers/inference.py

Browse files

Files changed (1) hide show

routers/inference.py +55 -18

routers/inference.py CHANGED Viewed

@@ -165,34 +165,58 @@ def load_sources_file(file_id: str) -> str:
 def extract_text_from_response(response):
     """
-    Extrai o texto da resposta de forma robusta.
     """
-    response_text = ""
     if hasattr(response, 'text') and response.text:
         return response.text
     if hasattr(response, 'candidates') and response.candidates:
-        for candidate in response.candidates:
-            if not hasattr(candidate, 'content') or not candidate.content:
-                continue
-            content = candidate.content
-            if not hasattr(content, 'parts') or content.parts is None:
-                continue
-            try:
-                parts_list = list(content.parts) if content.parts else []
-                for part in parts_list:
-                    if hasattr(part, 'text') and part.text:
-                        response_text += part.text
-            except Exception:
-                continue
-    return response_text
 def extract_sources_from_response(response):
     """
@@ -373,12 +397,23 @@ Daniels serves as a director on the show alongside Ken Kwapis, Yana Gorskaya, Pa
             config=config
         )
         # Extrair texto e fontes
         response_text = extract_text_from_response(response)
         sources = extract_sources_from_response(response)
         # Verificar se o texto está vazio
         if not response_text or response_text.strip() == "":
             raise HTTPException(
                 status_code=500,
                 detail="Modelo não retornou conteúdo válido"
@@ -401,6 +436,8 @@ Daniels serves as a director on the show alongside Ken Kwapis, Yana Gorskaya, Pa
             else:
                 content = "Conteúdo não encontrado"
         return NewsResponse(title=title, subhead=subhead, content=content, sources=sources)
     except HTTPException:

 def extract_text_from_response(response):
     """
+    Extrai o texto da resposta de forma robusta com debug.
     """
+    logger.info(f"Tipo da resposta: {type(response)}")
+    # Método 1: Verificar se tem atributo 'text' diretamente
     if hasattr(response, 'text') and response.text:
+        logger.info("Texto extraído via response.text")
         return response.text
+    # Método 2: Verificar candidates
     if hasattr(response, 'candidates') and response.candidates:
+        logger.info(f"Encontrados {len(response.candidates)} candidates")
+        for i, candidate in enumerate(response.candidates):
+            logger.info(f"Processando candidate {i}")
+            # Verificar se tem content
+            if hasattr(candidate, 'content') and candidate.content:
+                content = candidate.content
+                logger.info(f"Candidate {i} tem content")
+                # Verificar se tem parts
+                if hasattr(content, 'parts') and content.parts:
+                    logger.info(f"Content tem {len(list(content.parts))} parts")
+                    response_text = ""
+                    for j, part in enumerate(content.parts):
+                        logger.info(f"Processando part {j}, tipo: {type(part)}")
+                        if hasattr(part, 'text') and part.text:
+                            logger.info(f"Part {j} tem texto: {len(part.text)} caracteres")
+                            response_text += part.text
+                    if response_text:
+                        return response_text
+    # Método 3: Tentar acessar via string conversion
+    try:
+        response_str = str(response)
+        logger.info(f"Resposta como string (primeiros 200 chars): {response_str[:200]}")
+        # Se a string contém as tags esperadas, retornar ela
+        if '<headline>' in response_str and '<body>' in response_str:
+            return response_str
+    except Exception as e:
+        logger.error(f"Erro ao converter resposta para string: {e}")
+    # Debug: Logar atributos da resposta
+    logger.info(f"Atributos da resposta: {dir(response)}")
+    return ""
 def extract_sources_from_response(response):
     """
             config=config
         )
+        logger.info("Resposta do modelo recebida com sucesso")
         # Extrair texto e fontes
         response_text = extract_text_from_response(response)
         sources = extract_sources_from_response(response)
+        logger.info(f"Texto extraído: {len(response_text) if response_text else 0} caracteres")
         # Verificar se o texto está vazio
         if not response_text or response_text.strip() == "":
+            logger.error("Texto extraído está vazio")
+            # Debug adicional: tentar logar a resposta crua
+            try:
+                logger.error(f"Resposta crua (primeiros 500 chars): {str(response)[:500]}")
+            except:
+                logger.error("Não foi possível converter resposta para string")
             raise HTTPException(
                 status_code=500,
                 detail="Modelo não retornou conteúdo válido"
             else:
                 content = "Conteúdo não encontrado"
+        logger.info(f"Processamento concluído com sucesso - Título: {title[:50]}...")
         return NewsResponse(title=title, subhead=subhead, content=content, sources=sources)
     except HTTPException: