Subindo arquivos113311

app.py

@@ -39,6 +39,20 @@ def calculate_glcm_contrast(image):
     
     return contrast
 
+# Função para analisar a textura e a temperatura de cor
+def analyze_image_properties(image):
+    # Análise de cor (média RGB)
+    image_rgb = cv2.cvtColor(np.array(image), cv2.COLOR_BGR2RGB)
+    avg_color_per_row = np.average(image_rgb, axis=0)
+    avg_color = np.average(avg_color_per_row, axis=0)
+    temperature = 'fria' if np.mean(avg_color) < 128 else 'quente'
+
+    # Análise de textura
+    texture_contrast = calculate_glcm_contrast(image)
+    texture = 'lisa' se texture_contrast < 100 else 'texturizada'
+
+    return temperature, texture
+
 # Função para descrever imagem usando BLIP
 def describe_image(image):
     processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
@@ -66,15 +80,18 @@ def process_image(image):
     # Análise de cor (média RGB)
     mean_rgb = np.mean(np.array(image), axis=(0, 1))
 
-    # Análise de textura
-    texture_contrast = calculate_glcm_contrast(image)
+    # Análise de textura e temperatura de cor
+    temperature, texture = analyze_image_properties(image)
 
     # Descrição da imagem
     description = describe_image(image)
     translated_description = translate_description(description)
 
+    # Construir a descrição final
+    final_description = f"{translated_description}. A textura é {texture} e a temperatura de cor é {temperature}."
+
     # Texto para voz
-    tts = gTTS(text=translated_description, lang='pt')
+    tts = gTTS(text=final_description, lang='pt')
     attempts = 0
     while attempts < 5:
         try:
@@ -89,7 +106,7 @@ def process_image(image):
                 raise e
 
     # Retornar imagem com detecções, descrição e áudio
-    return Image.fromarray(detected_image), translated_description, "output.mp3"
+    return Image.fromarray(detected_image), final_description, "output.mp3"
 
 # Carregar imagem de exemplo diretamente do código
 example_image_path = "example1.JPG"