Update app.py

app.py

@@ -5,6 +5,7 @@ from dotenv import load_dotenv
 import pandas as pd
 from transformers import Blip2Processor, Blip2ForConditionalGeneration
 import torch
+import time
 
 # Carrega variáveis de ambiente
 load_dotenv()
@@ -31,13 +32,22 @@ def identify_foods(image):
     
     return processor.decode(outputs[0], skip_special_tokens=True)
 
-def analyze_nutrition(image):
+def analyze_nutrition(image, progress=gr.Progress()):
     """Análise nutricional da imagem"""
     try:
-        # Identifica alimentos
+        # Status inicial
+        progress(0, desc="Iniciando análise...")
+        yield "Iniciando análise...", None, None, None, None, "processing"
+        time.sleep(1)  # Simula processamento
+        
+        # Identificação dos alimentos
+        progress(0.2, desc="Identificando alimentos...")
         identified_foods = identify_foods(image)
+        yield f"🔄 Identificando alimentos...\n\n{identified_foods}", None, None, None, None, "processing"
+        time.sleep(1)
         
-        # Simula análise nutricional
+        # Análise nutricional
+        progress(0.4, desc="Calculando valores nutricionais...")
         nutrients = {
             "calorias": 450,
             "proteinas": 25,
@@ -61,7 +71,18 @@ def analyze_nutrition(image):
             ["Fibras", f"{nutrients['fibras']:.1f}g"]
         ]
         
-        # Análise e recomendações
+        progress(0.6, desc="Gerando análise...")
+        yield (
+            f"🔄 Gerando análise...\n\n{identified_foods}",
+            table_data,
+            plot_data,
+            "🔄 Analisando valores...",
+            "🔄 Preparando recomendações...",
+            "processing"
+        )
+        time.sleep(1)
+        
+        # Análise
         analysis = []
         if nutrients['calorias'] > 800:
             analysis.append("⚠️ Alto valor calórico")
@@ -72,7 +93,8 @@ def analyze_nutrition(image):
         
         analysis_text = "\n".join(analysis) if analysis else "✅ Valores nutricionais adequados"
         
-        # Recomendações personalizadas
+        progress(0.8, desc="Gerando recomendações...")
+        # Recomendações
         recommendations = []
         if "Alto valor calórico" in analysis_text:
             recommendations.append("• Considere reduzir as porções")
@@ -83,32 +105,30 @@ def analyze_nutrition(image):
             
         recommendations_text = "\n".join(recommendations) if recommendations else "• Continue mantendo uma alimentação equilibrada!"
         
-        return (
-            f"### Alimentos Identificados\n{identified_foods}",
+        progress(1.0, desc="Concluído!")
+        # Resultado final
+        yield (
+            f"### 🍽️ Alimentos Identificados\n{identified_foods}",
             table_data,
             plot_data,
-            f"### Análise\n{analysis_text}",
-            f"### Recomendações\n{recommendations_text}",
-            "success"  # Status para UI
+            f"### 📊 Análise\n{analysis_text}",
+            f"### 💡 Recomendações\n{recommendations_text}",
+            "success"
         )
         
     except Exception as e:
-        return (
-            None, None, None, None,
-            f"Erro na análise: {str(e)}",
-            "error"  # Status para UI
+        yield (
+            "❌ Erro na identificação",
+            None,
+            None,
+            "❌ Erro na análise",
+            f"Erro: {str(e)}",
+            "error"
         )
 
 css = """
-.container {
-    max-width: 1200px;
-    margin: auto;
-    padding: 20px;
-}
-.header {
-    text-align: center;
-    margin-bottom: 30px;
-}
+.container { max-width: 1200px; margin: auto; padding: 20px; }
+.header { text-align: center; margin-bottom: 30px; }
 .result-box {
     border: 1px solid #ddd;
     border-radius: 8px;
@@ -116,6 +136,24 @@ css = """
     margin: 10px 0;
     background-color: #fff;
 }
+.processing {
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    padding: 20px;
+    background: #f7f7f7;
+    border-radius: 8px;
+    margin: 10px 0;
+}
+.status-progress {
+    padding: 10px;
+    margin: 10px 0;
+    border-radius: 4px;
+    font-weight: bold;
+}
+.status-error { background: #fee; color: #c00; }
+.status-success { background: #efe; color: #0c0; }
+.status-processing { background: #eef; color: #00c; }
 """
 
 # Interface Gradio
@@ -153,9 +191,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as iface:
         # Coluna da Direita - Resultados
         with gr.Column(scale=3):
             # Status
-            status = gr.Markdown(visible=False)
+            status = gr.Markdown(elem_classes="status-progress", visible=False)
             
-            with gr.Tabs():
+            with gr.Tabs() as tabs:
                 # Tab de Identificação
                 with gr.Tab("🔍 Identificação"):
                     with gr.Group(elem_classes="result-box"):