Última mejora etiquetas

app.py

@@ -40,7 +40,11 @@ def predict(audio_path_pred):
         label_class = id2label_class[predicted_class_ids_class]
         label_mapping = {0: 'Cansancio/Incomodidad', 1: 'Dolor', 2: 'Hambre', 3: 'Problemas para respirar'}
         label_class = label_mapping.get(predicted_class_ids_class, label_class)
-    return label_class
+    return f"""
+        <div style='text-align: center; font-size: 1.5em'>
+            <span style='display: inline-block; min-width: 300px;'>{label_class}</span>
+        </div>
+    """
 
 def predict_stream(audio_path_stream):
     with torch.no_grad():
@@ -70,14 +74,24 @@ def mostrar_decibelios(audio_path_stream, visual_threshold):
         status = "Prediciendo..."
     else:
         status = "Esperando..."
-    return f"<h3 style='text-align: center; font-size: 1.5em;'>{status} Decibelios: {db_level:.2f}</h3>"
+    return f"""
+        <div style='text-align: center; font-size: 1.5em'>
+            <span>{status}</span>
+            <span style='display: inline-block; min-width: 120px;'>Decibelios: {db_level:.2f}</span>
+        </div>
+    """
 
 def predict_stream_decib(audio_path_stream, visual_threshold):
     db_level = decibelios(audio_path_stream)
     if db_level > visual_threshold:
-        return display_prediction_stream(audio_path_stream)
+        prediction = display_prediction_stream(audio_path_stream)
     else:
-        return ""
+        prediction = ""
+    return f"""
+        <div style='text-align: center; font-size: 1.5em; min-height: 2em;'>
+            <span style='display: inline-block; min-width: 300px;'>{prediction}</span>
+        </div>
+    """
 
 def chatbot_config(message, history: list[tuple[str, str]]):
     system_message = "You are a Chatbot specialized in baby health and care."
@@ -239,7 +253,7 @@ with gr.Blocks(theme=my_theme) as demo:
         audio_stream.stream(
             mostrar_decibelios,
             inputs=[audio_stream, threshold_db],
-            outputs=gr.HTML(value="<h3 style='text-align: center; font-size: 1.5em;'>Esperando...</h3>")
+            outputs=gr.HTML()
         )
         audio_stream.stream(
             predict_stream_decib,