Update app.py

app.py

@@ -36,7 +36,7 @@ def identify_image(image):
         score = top_5.values[i].item()
         results.append(f"{label}: {score:.2f}")
     
-    return "\n".join(results)
+    return ", ".join(results)
 
 def analyze_image(image):
     overall_start_time = time.time()
@@ -62,15 +62,16 @@ def analyze_image(image):
     ax.set_ylim([0, 1])
     ax.axis('off')
     
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
-        plt.savefig(temp_file, format='png', bbox_inches='tight', pad_inches=0)
-        histogram_path = temp_file.name
+    histogram_path = "histogram.png"
+    plt.savefig(histogram_path, format='png', bbox_inches='tight', pad_inches=0)
     plt.close(fig)
     times['histogram'] = f"{time.time() - start_time:.3f} seconds"
 
     # 2. Edge detection
     start_time = time.time()
     edges = cv2.Canny(gray_image, 100, 200)
+    edges_path = "edges.png"
+    cv2.imwrite(edges_path, edges)
     times['edge_detection'] = f"{time.time() - start_time:.3f} seconds"
 
     # 3. Main color extraction (16 colors)
@@ -88,6 +89,8 @@ def analyze_image(image):
     
     # Create color palette
     color_palette = create_color_palette(colors)
+    palette_path = "palette.png"
+    cv2.imwrite(palette_path, color_palette)
     times['main_colors'] = f"{time.time() - start_time:.3f} seconds"
 
     # Calculate overall elapsed time
@@ -97,7 +100,7 @@ def analyze_image(image):
     # Prepare timing output
     timing_output = "\n".join([f"{k}: {v}" for k, v in times.items()])
 
-    return top_5_classes, histogram_path, edges, color_palette, timing_output
+    return top_5_classes, histogram_path, edges_path, palette_path, timing_output
 
 # Define Gradio interface
 interface = gr.Interface(
@@ -105,13 +108,14 @@ interface = gr.Interface(
     inputs=gr.Image(type="pil", label="Upload Image"),
     outputs=[
         gr.Textbox(label="Image Classification (Top 5)"),
-        gr.Image(type="filepath", label="Histogram"),
-        gr.Image(type="numpy", label="Edge Detection"),
-        gr.Image(type="numpy", label="Main Colors Palette"),
+        gr.File(label="Histogram"),
+        gr.File(label="Edge Detection"),
+        gr.File(label="Main Colors Palette"),
         gr.Textbox(label="Processing Times")
     ],
     title="Advanced Image Analysis",
-    description="Upload an image to analyze its classification, histogram, edges, and main colors."
+    description="Upload an image to analyze its classification, histogram, edges, and main colors.",
+    allow_flagging="never"
 )
 
 # Launch the interface