from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import httpx
import random
import logging
import traceback

app = FastAPI()

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# 直接在代码中定义目标 URL 和 API Key
TARGET_URL = "https://ue-ai.excn.vip/api/v1/chat/completions"  # 替换为实际目标接口地址
API_KEY = "fastgpt-tkMWhPfR4tLmJVMA46KrA1zR89W1hbbV3Kcv4CYPVVUvvR5FsKDJ"  # 替换为实际的 API Key


class ImageRequest(BaseModel):
    images: dict[str, str]


def _id():
    return ''.join(random.choices('abcdefghijklmnopqrstuvwxyz', k=7))


@app.post("/process-images")
async def process_images(request: ImageRequest):
    image_data = request.images
    results = []

    # 设置最长等待响应时间为 500 秒
    timeout = httpx.Timeout(500.0, connect=10.0)

    async with httpx.AsyncClient(timeout=timeout) as client:
        for unique_id, image_url in image_data.items():
            payload = {
                "chatId": _id(),
                "stream": False,
                "messages": [
                    {
                        "role": "user",
                        "content": [
                            {"type": "image_url", "image_url": {"url": image_url}}
                        ]
                    }
                ]
            }

            try:
                response = await client.post(
                    url=TARGET_URL,
                    headers={
                        "Authorization": f"Bearer {API_KEY}",
                        "Content-Type": "application/json"
                    },
                    json=payload
                )
                response.raise_for_status()

                # Log the entire response content for debugging
                logger.info(f"Response content: {response.content}")

                try:
                    response_data = response.json()
                except ValueError:
                    logger.error("Failed to decode JSON from response")
                    logger.error(f"Non-JSON response content: {response.text}")
                    raise HTTPException(status_code=500, detail="Response is not valid JSON")

                # Extract the relevant message content from the response
                result_message = None
                if "choices" in response_data and len(response_data["choices"]) > 0:
                    message_content = response_data["choices"][0]["message"]["content"]
                    result_message = message_content

                results.append({"id": unique_id, "result": result_message})
                logger.info(f"Processed image {unique_id} successfully.")

            except httpx.ReadTimeout:
                logger.error("Request timed out.")
                raise HTTPException(status_code=504, detail="Request to target service timed out.")
            except httpx.HTTPStatusError as e:
                logger.error(f"HTTP error occurred: {str(e)}")
                logger.error(f"Response status code: {e.response.status_code}, content: {e.response.text}")
                raise HTTPException(status_code=e.response.status_code, detail=str(e))
            except Exception as e:
                logger.error(f"Unexpected error occurred: {str(e)}")
                logger.error(traceback.format_exc())  # Log the stack trace
                raise HTTPException(status_code=500, detail="An unexpected error occurred")

    return {"results": results}