Update app.py

app.py

@@ -301,7 +301,7 @@ def create_pdf(images, description, email, company):
     c.save()
 
     
-    return filename
+    return generate_qr_code(filename)
 
 
 # Function to generate image description using Claude 3 Sonnet
@@ -370,26 +370,32 @@ 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")
+def generate_qr_code(filename):
+    try:
+        # Read the content of the file
+        with open(filename, 'rb') as file:
+            file_content = file.read()
+
+        # Generate QR code
+        qr = qrcode.QRCode(
+            version=1,
+            error_correction=qrcode.constants.ERROR_CORRECT_L,
+            box_size=10,
+            border=4,
+        )
+        qr.add_data(file_content)
+        qr.make(fit=True)
 
-    # 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()
+        img_qr = qr.make_image(fill_color="black", back_color="white")
 
-    return img_byte_arr
+        # Convert PIL Image to bytes for Gradio output
+        img_byte_arr = io.BytesIO()
+        img_qr.save(img_byte_arr, format='PNG')
+        img_byte_arr = img_byte_arr.getvalue()
 
+        return img_byte_arr
+    except Exception as e:
+        return f"Error: {e}"
 
 with gr.Blocks() as demo:
 
@@ -427,22 +433,14 @@ 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=pdf_file
+                outputs=gr.Image(type="numpy", label="Generated QR Code")
             )
 
-            gr.Interface(
-            generate_qr_code,
-            inputs=pdf_file,
-            outputs=gr.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'))