Add Pillow dependency and implement screenshot functionality in web server
This commit is contained in:
		@@ -1,6 +1,8 @@
 | 
			
		||||
from flask import Flask, request, jsonify
 | 
			
		||||
import os, ai.processor
 | 
			
		||||
from dotenv import load_dotenv
 | 
			
		||||
import io
 | 
			
		||||
from PIL import ImageGrab
 | 
			
		||||
 | 
			
		||||
load_dotenv()
 | 
			
		||||
 | 
			
		||||
@@ -36,6 +38,14 @@ class WebServerApp:
 | 
			
		||||
                    img_data = img_file.read()
 | 
			
		||||
                else:
 | 
			
		||||
                    img_data = None
 | 
			
		||||
            else:
 | 
			
		||||
                if 'host_screenshot' in data:
 | 
			
		||||
                    # take a screenshot right here
 | 
			
		||||
                    # capture the full screen
 | 
			
		||||
                    screenshot_img = ImageGrab.grab()
 | 
			
		||||
                    buf = io.BytesIO()
 | 
			
		||||
                    screenshot_img.save(buf, format='PNG')
 | 
			
		||||
                    img_data = buf.getvalue() 
 | 
			
		||||
            
 | 
			
		||||
            import base64
 | 
			
		||||
            # Convert image data to base64 if provided
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user