Add Plot

app.py

@@ -43,18 +43,18 @@ def server(input: Inputs, output: Outputs, session: Session):
         return None
 
     @output
-    @render.image
-    def image_output():
-        here = Path(__file__).parent
-        if input.image_input():
-            src = input.image_input()[0]['datapath']
-            img = {"src": src, "width": "500px"} 
-            x = run.pred(src)
-            print(x)
-            return img
-        return None
-
-    @render.plot(alt="A histogram")  
+    # @render.image
+    # def image_output():
+    #     here = Path(__file__).parent
+    #     if input.image_input():
+    #         src = input.image_input()[0]['datapath']
+    #         img = {"src": src, "width": "500px"} 
+    #         x = run.pred(src)
+    #         print(x)
+    #         return img
+    #     return None
+
+    @render.plot 
     def plot(): 
         if input.image_input():
             new_image = input.image_input()[0]['datapath']
@@ -78,7 +78,10 @@ def server(input: Inputs, output: Outputs, session: Session):
                 ax.set_yticks([])
                 ax.set_xticklabels([])
                 ax.set_yticklabels([])
-
+        else:
+            fig, _ = plt.subplots()  # Create an empty figure if no image received
+            fig.text(0.5, 0.5, "No image received", ha='center', va='center', fontsize=14)
+            print("no image received")
         return fig