Update app.py

app.py

@@ -200,11 +200,11 @@ async def handle_text(task_type, prompt):
         return {"error": str(e)}
 
 async def handle_image(prompt):
-    """Handle image generation and quality analysis."""
+    """Handle image generation and quality analysis. Returns (image, json_data)."""
     global last_task_category
     last_task_category = "image"
     if image_pipe is None:
-        return {"error": "Image model not loaded"}, None
+        return None, {"error": "Image model not loaded"}
     try:
         import time
         start = time.time()
@@ -233,15 +233,15 @@ async def handle_image(prompt):
             latency_ms=gen_time * 1000
         )
         quality_result = await image_quality_detector.analyze(event)
-        return {
-            "image": image,
+        json_data = {
             "generation_time": gen_time,
             "retrieval_score": retrieval_score,
             "quality_detection": quality_result
         }
+        return image, json_data
     except Exception as e:
         logger.error(f"Image task error: {e}")
-        return {"error": str(e)}, None
+        return None, {"error": str(e)}
 
 async def handle_audio(audio_file):
     """Handle audio transcription and quality analysis."""
@@ -302,10 +302,10 @@ async def read_iot_sensors(fault_type):
     if len(iot_history) > 100:
         iot_history.pop(0)
 
-    # Create IoTEvent
+    # Create IoTEvent with valid component name (hyphens allowed, underscores not allowed)
     event = IoTEvent(
         timestamp=datetime.utcnow(),
-        component="robotic_arm",
+        component="robotic-arm",  # changed from 'robotic_arm'
         service_mesh="factory",
         latency_p99=0,
         error_rate=0.0,