Update app.py

app.py

@@ -14,20 +14,35 @@ import gradio as gr
 model = AutoModel.from_pretrained("Alibaba-NLP/gte-multilingual-base", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("Alibaba-NLP/gte-multilingual-base", trust_remote_code=True)
 
-# Example Gradio function
-def search_by_name(name):
-    # Your logic to search by name and return a response
-    pass
+# Function to load dataset (adjust this function if your dataset is complex)
+def load_dataset():
+    df = pd.read_excel("your_dataset.xlsx")  # Ensure this file exists in your working directory
+    print("Columns in the dataset:", df.columns.tolist())
+    return df
+
+# Example function to search by name and return the PEC number
+def search_by_name(name, df):
+    name_matches = df[df['Name'].str.contains(name, case=False, na=False)]
+    if not name_matches.empty:
+        return f"Your PEC number: {name_matches['PEC No'].values[0]}"
+    else:
+        return "No matches found for your name."
 
 # Gradio interface
-iface = gr.Interface(
-    fn=search_by_name,
-    inputs=gr.Textbox(label="Please write your Name"),
-    outputs=gr.Textbox(label="Your PEC number"),
-    title="PEC Number Lookup",
-    description="Enter your name to find your PEC number."
-)
+def build_interface():
+    df = load_dataset()  # Load your dataset
+    iface = gr.Interface(
+        fn=lambda name: search_by_name(name, df),
+        inputs=gr.Textbox(label="Please write your Name"),
+        outputs=gr.Textbox(label="Your PEC number"),
+        title="PEC Number Lookup",
+        description="Enter your name to find your PEC number."
+    )
+    return iface
+
+# Main function to run the Gradio app
+if __name__ == "__main__":
+    iface = build_interface()
+    iface.launch()
 
-# Launch the interface
-iface.launch()