Update app.py

app.py

@@ -27,26 +27,39 @@ def load_dataset():
         print(f"Error loading file: {e}")
         return None
 
-# Function to get details based on the name
-def get_details(name, df):
+# Function to get details based on the name or phone number
+def get_details(name, phone_number, df):
     # Clean the 'Name' column for consistent searching
     df['Name'] = df['Name'].astype(str).str.strip().str.lower()
-    name = name.strip().lower()
+    df['Contact Number (Whatsapp)'] = df['Contact Number (Whatsapp)'].astype(str).str.strip()
 
-    print(f"Searching for Name: {name}")  # Debugging output
-    results = df[df['Name'].str.contains(name, case=False, na=False)]
+    name = name.strip().lower() if name else None
+    phone_number = phone_number.strip() if phone_number else None
+
+    print(f"Searching for Name: {name}, Phone Number: {phone_number}")  # Debugging output
+
+    # Filter based on the provided inputs
+    if name and phone_number:
+        results = df[(df['Name'].str.contains(name, case=False, na=False)) |
+                     (df['Contact Number (Whatsapp)'] == phone_number)]
+    elif name:
+        results = df[df['Name'].str.contains(name, case=False, na=False)]
+    elif phone_number:
+        results = df[df['Contact Number (Whatsapp)'] == phone_number]
+    else:
+        return None  # If neither is provided, return None
 
     if not results.empty:
         print(f"Found Details: {results.to_dict('records')}")  # Debugging output
         return results.to_dict('records')[0]  # Return the first match for simplicity
     else:
-        print("Name not found.")  # Debugging output
+        print("No matching records found.")  # Debugging output
         return None
 
 # Function to format the output in HTML
 def format_output(details):
     if details is None:
-        return "Name not found."
+        return "No matching records found."
 
     # Build HTML table for a cleaner presentation
     html = "<table style='border-collapse: collapse; width: 100%;'>"
@@ -66,17 +79,20 @@ def add_or_update_details(name, phone_number, current_occupation, pass_out_year)
         if df is None:
             return "Error loading data."
 
-        # Find the entry for the name
-        details = get_details(name, df)
+        # Find the entry for the name or phone number
+        details = get_details(name, phone_number, df)
 
         if details:
             # Update the phone number, current occupation, and pass out year
-            df.loc[df['Name'].str.lower() == name.strip().lower(), 'Contact Number (Whatsapp)'] = phone_number
-            df.loc[df['Name'].str.lower() == name.strip().lower(), 'Current Occupation'] = current_occupation
-            df.loc[df['Name'].str.lower() == name.strip().lower(), 'Pass Out Year'] = pass_out_year
+            if phone_number:
+                df.loc[df['Name'].str.lower() == name.strip().lower(), 'Contact Number (Whatsapp)'] = phone_number
+            if current_occupation:
+                df.loc[df['Name'].str.lower() == name.strip().lower(), 'Current Occupation'] = current_occupation
+            if pass_out_year:
+                df.loc[df['Name'].str.lower() == name.strip().lower(), 'Pass Out Year'] = pass_out_year
             message = "Phone number, occupation, and pass out year updated successfully."
         else:
-            # Add a new entry if the name was not found
+            # Add a new entry if the name or phone number was not found
             new_entry = {
                 'Name': name,
                 'Contact Number (Whatsapp)': phone_number,
@@ -103,8 +119,8 @@ def predict_and_update(name, phone_number=None, current_occupation=None, pass_ou
         if df is None:
             return "Error loading the file."
 
-        # Get details for the provided name
-        details = get_details(name, df)
+        # Get details for the provided name or phone number
+        details = get_details(name, phone_number, df)
         details_html = format_output(details)
 
         # If a phone number, current occupation, or pass out year is provided, update or add the details
@@ -121,16 +137,19 @@ def predict_and_update(name, phone_number=None, current_occupation=None, pass_ou
 iface = gr.Interface(
     fn=predict_and_update,
     inputs=[
-        gr.Textbox(lines=1, label="**Enter Name**"),
-        gr.Textbox(lines=1, label="**Phone Number** (Optional - to add/update)"),
+        gr.Textbox(lines=1, label="**Enter Name** (Optional)"),
+        gr.Textbox(lines=1, label="**Phone Number** (Optional)"),
         gr.Textbox(lines=1, label="**Current Occupation** (Optional - to add/update)"),
         gr.Textbox(lines=1, label="**Pass Out Year** (Optional - to add/update)")
     ],
     outputs="html",
-    title="Name Details Lookup and Update",
-    description="Enter a name to find all details associated with it from the 'cleaned_data.xlsx' file. Optionally, add or update the phone number, current occupation, and pass out year."
+    title="Name and Phone Number Lookup and Update",
+    description="Enter a name or phone number to find details associated with it from the 'cleaned_data.xlsx' file. Optionally, add or update the phone number, current occupation, and pass out year."
 )
 
 # Run the Gradio interface
 if __name__ == "__main__":
     iface.launch()
+
+      
+