Adding root

main.py

@@ -205,7 +205,7 @@ def health():
         "status": "ok",
         "yolo_classes": len(yolo_model.names),
         "style_classes": len(STYLES),
-        "llm": "phi3_mini_4k_q4.gguf"
+        "llm": "qwen2.5-1.5b-instruct-fp16.gguf"
     }
 
 @app.post("/detect")
@@ -246,4 +246,32 @@ async def detect(file: UploadFile = File(...)):
         "annotated_image": encode_image(results.plot()),
         "image_width": int(img_np.shape[1]),
         "image_height": int(img_np.shape[0]),
-    })
+    })
+
+@app.get("/")
+def root():
+    return {
+        "status": "ok",
+        "description": "Art analysis API: detects objects, classifies artistic style and generates a Shakespearean sonnet from paintings.",
+        "models": {
+            "object_detection": {
+                "file": "yolo_deart.onnx",
+                "description": "Custom YOLOv8 fine-tuned for art object detection"
+            },
+            "style_classification": {
+                "file": "style_classifier.onnx",
+                "classes": len(STYLES),
+                "description": f"ONNX classifier for {len(STYLES)} artistic styles (Baroque, Impressionism, Cubism…)"
+            },
+            "poem_generation": {
+                "model": "Qwen2.5-1.5B-Instruct (FP16 GGUF)",
+                "backend": "llama-cpp",
+                "description": "Generates a Shakespearean sonnet based on detected objects and art style"
+            }
+        },
+        "endpoints": {
+            "GET  /":        "This overview",
+            "GET  /health":  "Runtime status and loaded model info",
+            "POST /detect":  "Upload a painting image → returns detections, style, poem and annotated image"
+        }
+    }