Kvikontent commited on
Commit
64af8cf
·
verified ·
1 Parent(s): 475db54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -10
app.py CHANGED
@@ -11,26 +11,20 @@ hf_token = os.environ.get("hf_token")
11
 
12
  # Function to generate a QR code with transparent fill color
13
  def generate_qr_image(url, image_size):
14
- # Determine box size based on the larger dimension of the image
15
- max_dim = max(image_size[0], image_size[1])
16
- box_size = max_dim // 25
17
- border = 1
18
-
19
  qr = qrcode.QRCode(
20
  version=1,
21
  error_correction=qrcode.constants.ERROR_CORRECT_L,
22
- box_size=box_size,
23
- border=border,
24
  )
25
  qr.add_data(url)
26
  qr.make(fit=True)
27
 
28
- qr_img = qr.make_image(fill_color="black", back_color=None).convert("RGBA")
29
  qr_img = qr_img.resize(image_size)
30
 
31
  return qr_img
32
 
33
-
34
  # Function to query the image from the Hugging Face API
35
  def query_image(text):
36
  API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
@@ -46,7 +40,7 @@ def generate_image(url, text):
46
  api_image = Image.open(io.BytesIO(image_bytes))
47
 
48
  # Generate QR code image
49
- qr_image = generate_qr_image(url, api_image.size)
50
 
51
  # Create a blank image with transparency
52
  final_image = Image.new('RGBA', api_image.size, (255, 255, 255, 0))
 
11
 
12
  # Function to generate a QR code with transparent fill color
13
  def generate_qr_image(url, image_size):
 
 
 
 
 
14
  qr = qrcode.QRCode(
15
  version=1,
16
  error_correction=qrcode.constants.ERROR_CORRECT_L,
17
+ box_size=50,
18
+ border=4,
19
  )
20
  qr.add_data(url)
21
  qr.make(fit=True)
22
 
23
+ qr_img = qr.make_image(fill_color=None, back_color="white").convert("RGBA")
24
  qr_img = qr_img.resize(image_size)
25
 
26
  return qr_img
27
 
 
28
  # Function to query the image from the Hugging Face API
29
  def query_image(text):
30
  API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
 
40
  api_image = Image.open(io.BytesIO(image_bytes))
41
 
42
  # Generate QR code image
43
+ qr_image = generate_qr_image(url, (250, 250))
44
 
45
  # Create a blank image with transparency
46
  final_image = Image.new('RGBA', api_image.size, (255, 255, 255, 0))