removed all test image stuff

app.py

@@ -47,8 +47,8 @@ latest_data = {
     "status": "N/A",
     "update_time": "Waiting for data...",
     "image_url": "N/A",
-    "progress": 0,  
-    "message": "",  
+    "progress": 0,
+    "message": "",
 }
 
 bambu_client = None
@@ -98,52 +98,57 @@ def bambu_on_message(client, userdata, message):
 
 def rpi_on_message(client, userdata, msg):
     global latest_data
-    
+
     try:
         payload = msg.payload.decode("utf-8")
         logger.info(f"Received message from RPI: {payload}")
-        
+
         data = json.loads(payload)
         status = data.get("status", "Unknown")
-        
+
         latest_data["status"] = status
-        latest_data["update_time"] = data.get("update_time", time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
-        
+        latest_data["update_time"] = data.get(
+            "update_time", time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
+        )
+
         if "nozzle_temperature" in data:
             latest_data["nozzle_temperature"] = data["nozzle_temperature"]
-        
+
         if "bed_temperature" in data:
             latest_data["bed_temperature"] = data["bed_temperature"]
-        
+
         if "error" in data:
             logger.error(f"Error from RPI: {data['error']}")
-       
+
         if status == "Ready":
             latest_data["progress"] = 0
             latest_data["message"] = "Printer ready"
         elif status == "Processing":
             latest_data["progress"] = 25
             latest_data["message"] = "Processing G-code..."
-        
+
     except Exception as e:
         logger.error(f"Error processing message from RPI: {e}")
-        
+
         try:
             result = json.loads(payload)
-            
+
             if "status" in result:
                 status = result["status"]
                 latest_data["status"] = status
- 
-            if "command" in result and result["command"] == "capture_image" and result.get("auto_triggered", False):
+
+            if (
+                "command" in result
+                and result["command"] == "capture_image"
+                and result.get("auto_triggered", False)
+            ):
                 logger.info("receive capture command")
-                threading.Thread(target=handle_auto_capture, args=(result,), daemon=True).start()
-                
+                threading.Thread(
+                    target=handle_auto_capture, args=(result,), daemon=True
+                ).start()
+
         except json.JSONDecodeError:
             logger.error(f"Invalid JSON in message: {payload}")
-            
-    except Exception as e:
-        logger.error(f"Error processing RPI message: {e}")
 
 
 def get_data(serial=DEFAULT_SERIAL):
@@ -202,11 +207,13 @@ def send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed):
                 json.dumps({"command": "generate_gcode", "parameters": params}),
             )
             logger.info("Parameters sent successfully to RPi for G-code generation")
-            
+
             global latest_data
             latest_data["status"] = "Sent"
-            latest_data["update_time"] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
-            
+            latest_data["update_time"] = time.strftime(
+                "%Y-%m-%d %H:%M:%S", time.localtime()
+            )
+
             return "Parameters sent successfully to RPi. Status: Sent"
         else:
             logger.warning("MQTT not connected, parameters not sent")
@@ -215,13 +222,11 @@ def send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed):
         logger.error(f"Error sending parameters: {e}")
         return f"Error sending parameters: {e}"
 
-      
     latest_data["status"] = "Sending"
     latest_data["progress"] = 10
     latest_data["message"] = "Sending parameters to printer..."
 
 
-
 def get_image_base64(image):
     if image is None:
         logger.warning("No image to encode")
@@ -244,21 +249,21 @@ def get_image_base64(image):
 def handle_auto_capture(message):
     try:
         logger.info(f"receive capture command: {message}")
-        
+
         print_job = message.get("print_job", "unknown_job")
         timestamp = message.get("timestamp", time.strftime("%Y-%m-%d %H:%M:%S"))
-        
+
         latest_data["auto_capture_requested"] = True
         latest_data["last_capture_job"] = print_job
         latest_data["last_capture_time"] = timestamp
-        
+
         return capture_image()
     except Exception as e:
         logger.error(f"error: {e}")
         return None, f"capture failed: {str(e)}"
 
 
-def capture_image(url=None, use_test_image=False, test_image_name=None):
+def capture_image(url=None):
     global rpi_client, latest_data
 
     if rpi_client is None:
@@ -289,14 +294,6 @@ def capture_image(url=None, use_test_image=False, test_image_name=None):
 
     url = latest_data["image_url"]
 
-    if use_test_image:
-        logger.info("Using test image instead of URL")
-        test_img = get_test_image(test_image_name)
-        if test_img:
-            return test_img
-        else:
-            logger.warning("Failed to get specified test image, trying URL")
-
     if url != "N/A":
         try:
             logger.info(f"Capturing image from URL: {url}")
@@ -310,16 +307,17 @@ def capture_image(url=None, use_test_image=False, test_image_name=None):
     else:
         raise Exception("url is 'N/A'")
 
+
 def update_print_status():
-   global latest_data
-   return (
-       latest_data["status"],
-       latest_data["nozzle_temperature"],
-       latest_data["bed_temperature"],
-       latest_data["update_time"],
-       latest_data.get("progress", 0),
-       latest_data.get("message", ""),
-   )
+    global latest_data
+    return (
+        latest_data["status"],
+        latest_data["nozzle_temperature"],
+        latest_data["bed_temperature"],
+        latest_data["update_time"],
+        latest_data.get("progress", 0),
+        latest_data.get("message", ""),
+    )
 
 
 def health_check():
@@ -420,13 +418,11 @@ with demo:
         logger.info("API call: get_data")
         return get_data()
 
-    def api_capture_frame(url=None, use_test_image=False, test_image_name=None):
-        logger.info(
-            f"API call: capture_frame with URL: {url}, use_test_image: {use_test_image}"
-        )
+    def api_capture_frame(url=None):
+        logger.info(f"API call: capture_frame with URL: {url}")
 
         try:
-            img = capture_image(url, use_test_image, test_image_name)
+            img = capture_image(url)
             if img:
                 if img.mode == "RGBA":
                     img = img.convert("RGB")
@@ -448,20 +444,14 @@ with demo:
         param_2=60,
         param_3=60,
         param_4=100,
-        use_test_image=False,
-        test_image_name=None,
     ):
         logger.info(
-            f"API call: lambda with params: {param_1}, {param_2}, {param_3}, {param_4}, use_test_image: {use_test_image}, test_image_name: {test_image_name}"
+            f"API call: lambda with params: {param_1}, {param_2}, {param_3}, {param_4}"
         )
         try:
             img = None
 
-            if use_test_image:
-                logger.info(f"Lambda using test image: {test_image_name}")
-                img = get_test_image(test_image_name)
-
-            elif (
+            if (
                 img_data
                 and isinstance(img_data, str)
                 and (img_data.startswith("http://") or img_data.startswith("https://"))
@@ -477,10 +467,6 @@ with demo:
                 except Exception as e:
                     logger.error(f"Failed to decode base64 image: {e}")
 
-            if img is None:
-                logger.info("No valid image data received, using default test image")
-                img = get_test_image()
-
             if img:
                 img_array = np.array(img)
 
@@ -575,8 +561,6 @@ with demo:
         fn=api_capture_frame,
         inputs=[
             gr.Textbox(label="Image URL"),
-            gr.Checkbox(label="Use Test Image", value=False),
-            gr.Textbox(label="Test Image Name", value=""),
         ],
         outputs=api_json_output,
         api_name="capture_frame",
@@ -590,8 +574,6 @@ with demo:
             gr.Number(label="Bed Temperature", value=60),
             gr.Number(label="Print Speed", value=60),
             gr.Number(label="Fan Speed", value=100),
-            gr.Checkbox(label="Use Test Image", value=False),
-            gr.Textbox(label="Test Image Name", value=""),
         ],
         outputs=api_json_output,
         api_name="lambda",