Update app.py

app.py

@@ -59,7 +59,7 @@ def format_output(details):
     return html
 
 # Function to add or update details
-def add_or_update_details(name, phone_number):
+def add_or_update_details(name, phone_number, current_occupation, pass_out_year):
     try:
         # Load the existing data
         df = load_dataset()
@@ -70,14 +70,18 @@ def add_or_update_details(name, phone_number):
         details = get_details(name, df)
 
         if details:
-            # Update the phone number
+            # Update the phone number, current occupation, and pass out year
             df.loc[df['Name'].str.lower() == name.strip().lower(), 'Contact Number (Whatsapp)'] = phone_number
-            message = "Phone number updated successfully."
+            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
+            message = "Phone number, occupation, and pass out year updated successfully."
         else:
             # Add a new entry if the name was not found
             new_entry = {
                 'Name': name,
                 'Contact Number (Whatsapp)': phone_number,
+                'Current Occupation': current_occupation,
+                'Pass Out Year': pass_out_year,
                 # Add other fields with default values as needed
             }
             df = df.append(new_entry, ignore_index=True)
@@ -92,7 +96,7 @@ def add_or_update_details(name, phone_number):
         return f"Error: {e}"
 
 # Combine the functions to create a prediction and update functionality
-def predict_and_update(name, phone_number=None):
+def predict_and_update(name, phone_number=None, current_occupation=None, pass_out_year=None):
     try:
         # Load the dataset from the provided file path
         df = load_dataset()
@@ -103,9 +107,9 @@ def predict_and_update(name, phone_number=None):
         details = get_details(name, df)
         details_html = format_output(details)
 
-        # If a phone number is provided, update or add the details
-        if phone_number:
-            update_message = add_or_update_details(name, phone_number)
+        # If a phone number, current occupation, or pass out year is provided, update or add the details
+        if phone_number or current_occupation or pass_out_year:
+            update_message = add_or_update_details(name, phone_number, current_occupation, pass_out_year)
             return f"{details_html}<br><br>{update_message}"
         else:
             return details_html
@@ -118,14 +122,15 @@ 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="**Phone Number** (Optional - to add/update)"),
+        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."
+    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."
 )
 
 # Run the Gradio interface
 if __name__ == "__main__":
     iface.launch()
-