Update app.py

app.py

@@ -173,7 +173,7 @@ demo = gr.Interface(
     theme=gr.themes.Soft()
 )"""
 
-import plotly.graph_objects as go
+"""import plotly.graph_objects as go
 
 def predire(image):
     image_resized = tf.image.resize(image, (224, 224)) / 255.0
@@ -202,6 +202,48 @@ def predire(image):
         height=500,
         font=dict(size=16)
     )
+
+
+    # Pour placer le texte à l’intérieur des barres
+    fig.show(config={'displayModeBar': False})
+    return fig
+"""
+
+import plotly.graph_objects as go
+
+def predire(image):
+    image_resized = tf.image.resize(image, (224, 224)) / 255.0
+    preds = model.predict(tf.expand_dims(image_resized, axis=0))[0]
+    sorted_indices = np.argsort(preds)[::-1]
+    sorted_classes = [classes[i] for i in sorted_indices]
+    sorted_probs = [preds[i] for i in sorted_indices]
+    colors = ['#2ecc71' if p >= 0.5 else '#bdc3c7' for p in sorted_probs]
+
+    fig = go.Figure(go.Bar(
+        x=sorted_classes,
+        y=sorted_probs,
+        marker=dict(color=colors, line=dict(color='black', width=1)),
+        text=[f"{p*100:.1f}%" for p in sorted_probs],
+        textposition='auto'
+    ))
+
+    fig.update_layout(
+        xaxis=dict(fixedrange=True, tickangle=45, tickfont=dict(size=15), automargin=True),
+        yaxis=dict(fixedrange=True, range=[0, 1], title="Probabilité", tickfont=dict(size=14)),
+        title=dict(
+            text="Probabilités par mouvement pictural",
+            y=0.94,
+            pad=dict(b=30)
+        ),
+        margin=dict(l=20, r=20, t=60, b=80),  # marge top plus haute et bottom plus grande
+        height=500,
+        font=dict(size=16)
+    )
+
+    # Pour placer le texte à l’intérieur des barres
+    fig.data[0].textposition = 'inside'
+    fig.data[0].insidetextanchor = 'middle'
+    fig.data[0].textfont = dict(color='black', size=14, family="Arial")
     fig.show(config={'displayModeBar': False})
     return fig