Update app.py

app.py

@@ -947,21 +947,27 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="cyan")) a
                     diag = go.Figure(layout={"title": "Error processing data"})
                     entropy = go.Figure(layout={"title": "Error processing data"})
 
-                primary_info = f"""
-                <b>Primary:</b> {primary_row['filepath']}<br>
-                <b>Species:</b> {primary_row['source']}<br>
-                <b>Label:</b> {primary_row.get('label', 'N/A')}<br>
-                <b>Alpha:</b> {primary_cmt['alpha']:.4f}<br>
-                <b>SRL:</b> {primary_cmt['srl']:.4f}
-                """ if primary_cmt else ""
+                if primary_cmt:
+                    primary_info = f"""
+                    <b>Primary:</b> {primary_row['filepath']}<br>
+                    <b>Species:</b> {primary_row['source']}<br>
+                    <b>Label:</b> {primary_row.get('label', 'N/A')}<br>
+                    <b>Alpha:</b> {primary_cmt['alpha']:.4f}<br>
+                    <b>SRL:</b> {primary_cmt['srl']:.4f}
+                    """
+                else:
+                    primary_info = ""
                 
-                neighbor_info = f"""
-                <b>Neighbor:</b> {neighbor_row['filepath'] if neighbor_row is not None else 'N/A'}<br>
-                <b>Species:</b> {neighbor_row['source'] if neighbor_row is not None else 'N/A'}<br>
-                <b>Label:</b> {neighbor_row.get('label', 'N/A') if neighbor_row is not None else 'N/A'}<br>
-                <b>Alpha:</b> {neighbor_cmt['alpha']:.4f if neighbor_cmt else 0}<br>
-                <b>SRL:</b> {neighbor_cmt['srl']:.4f if neighbor_cmt else 0}
-                """ if neighbor_cmt else ""
+                if neighbor_cmt and neighbor_row is not None:
+                    neighbor_info = f"""
+                    <b>Neighbor:</b> {neighbor_row['filepath']}<br>
+                    <b>Species:</b> {neighbor_row['source']}<br>
+                    <b>Label:</b> {neighbor_row.get('label', 'N/A')}<br>
+                    <b>Alpha:</b> {neighbor_cmt['alpha']:.4f}<br>
+                    <b>SRL:</b> {neighbor_cmt['srl']:.4f}
+                    """
+                else:
+                    neighbor_info = ""
 
                 return dual_holo, diag, entropy, primary_info, neighbor_info