Update app.py

app.py

@@ -21,6 +21,7 @@ from reportlab.lib.pagesizes import letter
 from reportlab.lib.utils import ImageReader
 import boto3
 import re
+import qrcode
 
 output_parser = StrOutputParser()
 
@@ -242,7 +243,7 @@ def record_participant(email, company, description, images):
     participants_db.insert_one(participant_data)
 
     # Create PDF after recording participant
-    pdf_file = create_pdf(images, description, email, company)
+    qr_image = create_pdf(images, description, email, company)
     return pdf_file
 
 def create_pdf(images, description, email, company):
@@ -369,6 +370,26 @@ def start_image_search(image, text):
     description = generate_image_description_with_claude(images_base64_strs, img_base64_str)
     return images, description
 
+def generate_qr_code(url):
+    # Generate QR code
+    qr = qrcode.QRCode(
+        version=1,
+        error_correction=qrcode.constants.ERROR_CORRECT_L,
+        box_size=10,
+        border=4,
+    )
+    qr.add_data(url)
+    qr.make(fit=True)
+
+    img_qr = qr.make_image(fill_color="black", back_color="white")
+
+    # Convert PIL Image to bytes for displaying or saving
+    img_byte_arr = io.BytesIO()
+    img_qr.save(img_byte_arr, format='PNG')
+    img_byte_arr = img_byte_arr.getvalue()
+
+    return img_byte_arr
+
 
 with gr.Blocks() as demo:
 
@@ -406,11 +427,22 @@ with gr.Blocks() as demo:
                 outputs=[output_gallery, output_description]
             )
 
+            pdf_file=gr.File(label="Download Search Results as PDF")
+
             record_button.click(
                 fn=record_participant,
                 inputs=[email_input, company_input, output_description, output_gallery],
-                outputs=gr.File(label="Download Search Results as PDF")
+                outputs=pdf_file
             )
+
+            gr.Interface(
+            generate_qr_code,
+            inputs=pdf_file
+            outputs=gr.outputs.Image(type="numpy", label="Generated QR Code"),
+            title="Filename to QR Code Generator",
+            description="Enter the filename to generate its QR code"
+        )
+
         with gr.Tab("Code"):
             gr.Code(label="Code", language="python", value=fetch_url_data('https://huggingface.co/spaces/MongoDB/aws-bedrock-celeb-matcher/raw/main/app.py'))