Update app.py

app.py

@@ -1,67 +1,61 @@
-import pandas as pd
 import gradio as gr
+import pandas as pd
+import vobject
+from io import BytesIO
 
-def create_vcf(row):
-    vcf_template = """BEGIN:VCARD
-VERSION:3.0
-PRODID:-//Apple Inc.//iPhone OS 17.2.1//EN
-N:{name}
-ORG:BAT Bangladesh;{function}
-TITLE:{designation}
-TEL;type=CELL;type=VOICE;type=pref:{number}
-EMAIL:{email}
-END:VCARD"""
-    return vcf_template.format(
-        name=row['Name'],
-        function=row['Function'],
-        designation=row['Designation'],
-        number=row['Number'],
-        email=row['Email']
-    )
-
-def generate_vcf(file):
-    df = pd.read_excel(file)
-    df['Number'] = df['Number'].astype(str)  # Ensure phone numbers are strings
-    vcf_data = df.apply(create_vcf, axis=1).str.cat(sep='\n\n')
+def excel_to_vcf(file):
+    # Load Excel file
+    df = pd.read_excel(file.name)
     
-    vcf_file_name = '/tmp/BAT New Contacts.vcf'
-    with open(vcf_file_name, 'w') as vcf_file:
-        vcf_file.write(vcf_data)
-    
-    return vcf_file_name
+    # Prepare a dictionary to check for existing emails
+    contact_dict = {}
 
-# Define custom CSS and HTML
-css = """
-.gradio-container {
-    background: rgb(14, 43, 99);
-    display: flex;
-    flex-direction: column;
-    align-items: center;
-    color: #fff; /* Text color for better readability */
-    padding: 20px; /* Padding for the container */
-}
-.gradio-input, .gradio-output {
-    background: rgb(28, 56, 113);
-    color: #fff; /* Text color for input and output fields */
-}
-footer {
-    display: none !important;
-}
-"""
+    # Create a vCard file in memory
+    vcf_output = BytesIO()
+    
+    # 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'))
 
-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>
-"""
+    # Return the generated VCF file
+    vcf_output.seek(0)
+    return vcf_output, "contacts.vcf"
 
-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"),
-    )
+# 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."
+)
 
-demo.launch()
+iface.launch()