Update main.py

main.py

@@ -9,6 +9,8 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from typing import List, Dict, Any, Optional
 from datetime import datetime
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import StreamingResponse  # Add this import
 
 # Mock implementations for ImageResponse and to_data_uri
 class ImageResponse: