Update app.py

app.py

@@ -46,23 +46,25 @@ def get_name(pec_number, df):
         print("PEC Number not found.")  # Debugging output
         return "PEC Number not found."
 
-# Function to process the PEC number using the Hugging Face model
-def process_with_model(pec_number):
-    inputs = tokenizer(pec_number, return_tensors="pt")
-    with torch.no_grad():
-        outputs = model(**inputs)
-    # Instead of returning the full model output, return just a simple confirmation
-    return "Model processed successfully."
+# Function to check if the PEC number is attached
+def check_pec_number(pec_number, df):
+    df['PEC No.'] = df['PEC No.'].str.strip().str.upper()
+    pec_number = pec_number.strip().upper()
+
+    if pec_number in df['PEC No.'].values:
+        return "Your PEC Number is Attached."
+    else:
+        return "Your PEC Number is Not Attached."
 
-# Combine both functions to create a prediction
+# Combine the functions to create a prediction
 def predict(pec_number):
     try:
         # Load the dataset from the root directory
         df = load_dataset()
 
         name = get_name(pec_number, df)
-        model_output = process_with_model(pec_number)
-        return f"Name: {name}"  # Only display the name
+        pec_status = check_pec_number(pec_number, df)
+        return f"Your Name is: {name}\n{pec_status}"  # Return name and PEC status
     except Exception as e:
         print(f"An error occurred: {e}")
         return f"Error: {e}"
@@ -70,10 +72,10 @@ def predict(pec_number):
 # Build the Gradio interface without the file upload option
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Textbox(lines=1, placeholder="Enter PEC Number..."),
+    inputs=gr.Textbox(lines=1, placeholder="**Enter PEC Number...**"),  # Bold placeholder text
     outputs="text",
     title="PEC Number to Name Lookup",
-    description="Enter a PEC number to retrieve the corresponding name."
+    description="Enter a PEC number to retrieve the corresponding name and check if your PEC number is attached."
 )
 
 # Run the Gradio interface