Update main.py

main.py

@@ -1,65 +1,20 @@
 import streamlit as st
-import pandas as pd
-from io import BytesIO
-from loading_file import precomputed_df, generate_pdf
+from loading_file import fetch_data
+from search import search_application
 
-# Display a message with a link in markdown format
-st.write("If this application is down, please move to this link:")
+# Fetch data and store it in df
+df = fetch_data()
 
-# Button that opens the link when clicked
-if st.button("Go to backup link"):
-    st.write("Opening backup link...")
-    st.markdown('<a href="https://ireland.streamlit.app/" target="_blank">Click here</a>', unsafe_allow_html=True)
+# Title for the Streamlit app (Move this to the top)
+st.title("Visa Application Status Checker")
 
-# Title of the app
-st.title("Visa Application Search")
-
-# Debugging: Show Data Preview
-if not precomputed_df.empty:
-    st.write("### Debugging: Data Loaded")
-    st.dataframe(precomputed_df.head(10))  # Display first 10 rows
-
-# **Search Application by Number**
+# Streamlit UI to input the application number
 application_number = st.text_input("Enter Application Number:")
-search_button = st.button("Search")
 
-if search_button:
-    # Perform search
-    if application_number and application_number.strip():
-        try:
-            application_number = int(application_number.strip())  # Convert input to integer
-            result_df = precomputed_df[precomputed_df["Application Number"] == application_number]
-            
-            if not result_df.empty:
-                st.write("### Search Result:")
-                st.dataframe(result_df)
-            else:
-                st.warning("No results found.")
-        except ValueError:
-            st.warning("Please enter a valid numeric application number.")
+# Add a search button that triggers the search only when clicked
+if st.button("Search"):
+    if application_number:
+        # Call the search_application function with both application number and df
+        search_application(application_number, df)
     else:
-        st.warning("Please enter a valid application number.")
-
-# **Place Download Buttons BELOW the Search Button**
-if not precomputed_df.empty:
-    st.write("## Download Full Decision List")
-    
-    # Excel Download
-    excel_buffer = BytesIO()
-    precomputed_df.to_excel(excel_buffer, index=False, engine="openpyxl")
-    excel_buffer.seek(0)
-    st.download_button(
-        label="Download Excel File",
-        data=excel_buffer,
-        file_name="Visa_Decisions.xlsx",
-        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-    )
-
-    # PDF Download
-    pdf_buffer = generate_pdf(precomputed_df, "Visa Decisions")
-    st.download_button(
-        label="Download PDF File",
-        data=pdf_buffer,
-        file_name="Visa_Decisions.pdf",
-        mime="application/pdf",
-    )
+        st.warning("Please enter an Application Number to search.")