Update main.py

main.py

@@ -1,59 +1,36 @@
 import streamlit as st
-from loading_file import df  # Load and preprocess data
-from dataframe import precomputed_df  # Get precomputed nearest applications
-from search import search_application  # Search functionality
+from search import search_application
+from dataframe import precomputed_df
 
-# Set the page title
-st.set_page_config(page_title="Visa Application Status Checker", layout="wide")
-
-# Application title
 st.title("Visa Application Status Checker")
 
-# Navigation menu for modular structure
-menu = ["Home", "Search Application", "About"]
-choice = st.sidebar.selectbox("Menu", menu)
-
-if choice == "Home":
-    st.subheader("Welcome to the Visa Application Status Checker!")
-    st.write(
-        """
-        This application allows you to:
-        - Check the status of your Visa Application.
-        - Find the nearest application records for invalid application numbers.
-        - Explore real-time Visa updates.
-        """
-    )
-    st.write("Navigate to 'Search Application' to start.")
+# User input for application number
+application_number = st.text_input("Enter your Application Number (include 'IRL' if applicable):")
+search_button = st.button("Search")
 
-elif choice == "Search Application":
-    st.subheader("Search Your Application Status")
-    
-    # Input field for Application Number
-    user_input = st.text_input("Enter your Application Number (including 'IRL'):")
-    search_button = st.button("Search")  # Search button
+if search_button and application_number:
+    try:
+        # Parse the application number
+        if "IRL" in application_number.upper():
+            application_number = int("".join(filter(str.isdigit, application_number.upper().split("IRL")[-1])))
+        else:
+            application_number = int(application_number)
 
-    if search_button:
-        if not user_input.isdigit() or len(user_input) > 8:
-            st.warning("Please enter a valid Application Number with up to 8 digits.")
+        # Validate input length
+        if len(str(application_number)) > 8:
+            st.warning("Please enter a valid application number with up to 8 digits.")
         else:
-            user_input = int(user_input)
-            search_result = search_application(user_input)
+            result = search_application(application_number)
 
-            if search_result is not None:
-                decision = search_result["Decision"]
-                if decision.lower() == "approved":
-                    st.success(f"Congratulations! Your application ({user_input}) is Approved.")
-                elif decision.lower() == "refused":
-                    st.error(f"Your application ({user_input}) is Refused.")
-                else:
-                    st.info(f"Your application ({user_input}) has a status of '{decision}'.")
+            if result is not None and isinstance(result, pd.Series):
+                # Display exact match
+                st.subheader("Exact Match Found")
+                st.success(f"Application Number: {result['Application Number']}, Decision: {result['Decision']}")
+            elif result is not None:
+                # Display nearest records
+                st.subheader("Nearest Application Numbers")
+                st.table(result[["Application Number", "Decision", "Difference"]])
             else:
-                st.warning(f"No record found for Application Number {user_input}.")
-else:
-    st.subheader("About")
-    st.write(
-        """
-        This application was built using **Streamlit** to help users check their Visa application status.
-        Data is updated in real-time and fetched directly from the official website.
-        """
-    )
+                st.warning(f"No records found for Application Number {application_number}.")
+    except ValueError:
+        st.error("Invalid input. Please enter numeric digits only.")