Update app.py

app.py

@@ -1,61 +1,76 @@
-import gradio as gr
 import pandas as pd
-import vobject
-from io import BytesIO
+import gradio as gr
+
+# Function to create a vCard entry from a DataFrame row
+def create_vcf(row):
+    vcf_template = """BEGIN:VCARD
+VERSION:3.0
+PRODID:-//Apple Inc.//iPhone OS 17.2.1//EN
+N:{name}
+FN:{name}
+ORG:BAT Bangladesh;{function}
+TITLE:{designation}
+TEL;TYPE=CELL;TYPE=VOICE;TYPE=pref:{number}
+EMAIL:{email}
+END:VCARD"""
+    # Formatting the row data into vCard format
+    return vcf_template.format(
+        name=row['Name'],
+        function=row['Function'],
+        designation=row['Designation'],
+        number=row['Number'],
+        email=row['Email']
+    )
 
-def excel_to_vcf(file):
+# Function to generate the VCF file from the uploaded Excel data
+def generate_vcf(file):
     # Load Excel file
-    df = pd.read_excel(file.name)
+    df = pd.read_excel(file)
+    df['Number'] = df['Number'].astype(str)  # Convert phone numbers to string to avoid formatting issues
     
-    # Prepare a dictionary to check for existing emails
-    contact_dict = {}
-
-    # Create a vCard file in memory
-    vcf_output = BytesIO()
+    # Combine each vCard entry into a single VCF file content
+    vcf_data = df.apply(create_vcf, axis=1).str.cat(sep='\n\n')
+    
+    # Write the VCF data to a file
+    vcf_file_name = '/tmp/BAT_New_Contacts.vcf'
+    with open(vcf_file_name, 'w') as vcf_file:
+        vcf_file.write(vcf_data)
     
-    # Loop through each row in the DataFrame
-    for _, row in df.iterrows():
-        name = row['Name']
-        designation = row['Designation']
-        function = row['Function']
-        grade = row['GRADE']
-        email = row['Email']
-        number = str(row['Number'])
-        
-        # If email already exists, replace the contact details
-        if email in contact_dict:
-            contact_dict[email]['tel'].value = number
-        else:
-            # Create a new vCard entry
-            contact = vobject.vCard()
-            contact.add('fn').value = name
-            contact.add('title').value = designation
-            contact.add('org').value = function
-            contact.add('note').value = f"GRADE: {grade}"
-            
-            email_entry = contact.add('email')
-            email_entry.value = email
-            email_entry.type_param = 'INTERNET'
-            
-            tel_entry = contact.add('tel')
-            tel_entry.value = number
-            tel_entry.type_param = 'CELL'
-            
-            # Add contact to dictionary and write to VCF
-            contact_dict[email] = contact
-            vcf_output.write(contact.serialize().encode('utf-8'))
+    return vcf_file_name
+
+# Custom CSS and HTML for a refined UI
+css = """
+.gradio-container {
+    background: rgb(14, 43, 99);
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    color: #fff;
+    padding: 20px;
+}
+.gradio-input, .gradio-output {
+    background: rgb(28, 56, 113);
+    color: #fff;
+}
+footer {
+    display: none !important;
+}
+"""
 
-    # Return the generated VCF file
-    vcf_output.seek(0)
-    return vcf_output, "contacts.vcf"
+html_content = """
+<div style="text-align: center; margin-bottom: 20px;">
+    <img src="https://i.ibb.co/RbQRzcy/APMEA-CENTRAL-White.png" border="0" alt='BAT Bangladesh Logo' style='max-width: 300px;'>
+</div>
+<p style="text-align: center; color: #fff;">Upload an Excel file containing contact information in the following format: Name, Designation, Function, GRADE, Email, Number. The output will be a VCF file containing the contact information.</p>
+"""
 
-# Set up Gradio interface
-iface = gr.Interface(
-    fn=excel_to_vcf,
-    inputs=gr.inputs.File(label="Upload Excel File (Format: Name, Designation, Function, GRADE, Email, Number)"),
-    outputs=gr.outputs.File(label="Download VCF File"),
-    title="Excel to VCF Converter",
-    description="Upload an Excel file with contact information to generate a VCF file. Contacts with matching email IDs will have their phone numbers updated."
-)
+# Gradio Interface
+with gr.Blocks(css=css) as demo:
+    gr.HTML(html_content)
+    gr.Interface(
+        fn=generate_vcf,
+        inputs=gr.File(label="Upload Excel File"),
+        outputs=gr.File(label="Download VCF File"),
+    )
 
-iface.launch()
+demo.launch()