import cv2

def take_screenshot(cam, width=640, height=480):
    """
    Takes a screenshot of the primary display and returns it as a numpy array.
    """
    # Capture the screenshot using OpenCV
    print()
    "http://172.31.192.151/tmpfs/snap.jpg?usr=admin&pwd=admin"

import requests
import os
import time

# 请将此处的IP地址替换为您的ESP32在串口监视器中打印的实际IP地址
# 例如: esp32_ip = "192.168.1.100"
esp32_ip = "172.31.192.104" # <-- 替换为您的ESP32的IP地址

# 图片保存路径和文件名
output_directory = "captured_images"
output_filename = "esp32_camera_image.jpg"
output_filepath = os.path.join(output_directory, output_filename)

def fetch_image_from_esp32(ip_address, save_path):
    """
    从ESP32摄像头服务器获取图片并保存到指定路径。

    Args:
        ip_address (str): ESP32摄像头服务器的IP地址。
        save_path (str): 图片的保存路径，包括文件名。
    """
    url = f"http://{ip_address}/stream" # ESP32摄像头服务器通常在根路径提供图片

    print(f"尝试从 {url} 获取图片...")
    try:
        # 发送GET请求，设置超时时间
        response = requests.get(url, timeout=10)

        # 检查响应状态码
        if response.status_code == 200:
            # 确保保存目录存在
            os.makedirs(os.path.dirname(save_path), exist_ok=True)

            # 将图片内容写入文件
            with open(save_path, 'wb') as f:
                f.write(response.content)
            print(f"图片成功保存到: {save_path}")
        else:
            print(f"从 {url} 获取图片失败。状态码: {response.status_code}")
            print(f"响应内容: {response.text[:200]}...") # 打印部分响应内容以便调试

    except requests.exceptions.ConnectionError as e:
        print(f"连接错误: 无法连接到ESP32 ({ip_address})。请确保ESP32已连接到WiFi并运行摄像头服务器。错误: {e}")
    except requests.exceptions.Timeout as e:
        print(f"请求超时: 从ESP32 ({ip_address}) 获取图片超时。错误: {e}")
    except requests.exceptions.RequestException as e:
        print(f"请求发生未知错误: {e}")
    except Exception as e:
        print(f"发生意外错误: {e}")

if __name__ == "__main__":
    if esp32_ip == "YOUR_ESP32_IP_ADDRESS":
        print("警告: 请将 'esp32_ip' 变量替换为您的ESP32的实际IP地址。")
        print("您可以在ESP32的串口监视器中找到这个IP地址，它通常在 'Camera Ready! Use http://...' 这一行显示。")
    else:
        fetch_image_from_esp32(esp32_ip, output_filepath)