update append legend

app.py

@@ -23,12 +23,18 @@ if len(uploaded_files) == 3:
 
     plt.figure(figsize=(10, 6))
     line_styles = ['-', '--', '-.'] 
-    colors = ['b', 'g', 'r'] 
-    for i, (line, file) in enumerate(zip(scan_lines, uploaded_files)):
-        plt.plot(line, label=f'{file.name}', linestyle=line_styles[i % len(line_styles)], color=colors[i % len(colors)], alpha=0.7, linewidth=1.5)
-        
+    colors = ['b', 'g', 'r']
+
+    for i, scan_line in enumerate(scan_lines):
+        file_name = uploaded_files[i].name
+        style = line_styles[i % len(line_styles)]
+        color = colors[i % len(colors)]
+
+        plt.plot(scan_line, label=file_name, linestyle=style, color=color, alpha=0.7, linewidth=1.5)
+
     plt.xlabel('Pixel Position')
     plt.ylabel('Pixel Value')
     plt.title('Comparison of Scan Lines Across Images')
     plt.legend(loc='upper left')
     st.pyplot(plt)
+