Update app.py

app.py

@@ -19,11 +19,24 @@ generator = load_model()
 # Függvény a dátumok felismerésére és konverziójára
 def parse_date(value):
     try:
+        # Ha az érték tartalmaz évet
+        if any(char.isdigit() for char in value):
+            return datetime.strptime(value, "%d %B %Y")
+        # Ha nincs év megadva, alapértelmezett 2025
         parsed_date = datetime.strptime(value, "%d %B")
         return parsed_date.replace(year=2025)
     except ValueError:
         return None
 
+# Adatok Excel exportálása
+@st.cache_data
+def convert_to_excel(data):
+    import io
+    output = io.BytesIO()
+    with pd.ExcelWriter(output, engine="openpyxl") as writer:
+        data.to_excel(writer, index=False)
+    return output.getvalue()
+
 # Fájl feltöltése
 uploaded_file = st.file_uploader("Tölts fel egy Excel fájlt", type=["xlsx"])
 
@@ -48,7 +61,12 @@ if uploaded_file:
                 st.write(f"Releváns adat található a '{sheet}' munkalapon.")
                 relevant_data.append(df)
             else:
-                st.write(f"A '{sheet}' munkalap nem tartalmaz releváns adatokat.")
+                st.warning(f"A '{sheet}' munkalap nem tartalmaz automatikusan felismerhető adatokat.")
+                name_column = st.selectbox(f"Válaszd ki a 'Név' oszlopot a '{sheet}' munkalapon:", df.columns)
+                availability_column = st.selectbox(f"Válaszd ki az 'Elérhetőség' oszlopot a '{sheet}' munkalapon:", df.columns)
+                restriction_column = st.selectbox(f"Válaszd ki a 'Korlátozások' oszlopot a '{sheet}' munkalapon:", df.columns)
+                df.rename(columns={name_column: "Név", availability_column: "Elérhetőség", restriction_column: "Korlátozások"}, inplace=True)
+                relevant_data.append(df)
 
         # Összesített adatok
         if relevant_data:
@@ -58,12 +76,13 @@ if uploaded_file:
 
             # Dátumok feldolgozása az elérhetőség oszlopban
             combined_data["Elérhető napok"] = combined_data["Elérhetőség"].apply(
-                lambda x: [parse_date(day.strip()) for day in str(x).split(",") if isinstance(x, str)] if pd.notna(x) else []
+                lambda x: [parse_date(day.strip()) for day in str(x).split(",")] if pd.notna(x) else []
             )
             combined_data["Korlátozások"] = combined_data["Korlátozások"].apply(
                 lambda x: x.split(",") if isinstance(x, str) else []
             )
 
+            # Felhasználói input: dátumok kiválasztása
             available_days = st.multiselect(
                 "Add meg az elérhetőségi dátumokat:",
                 options=[day.strftime("%Y-%m-%d") for day in combined_data["Elérhető napok"] if day is not None]
@@ -91,27 +110,35 @@ if uploaded_file:
                                 })
                                 break
 
+                # Beosztás megjelenítése
                 beosztas_df = pd.DataFrame(beosztas)
                 st.write("Generált Ügyeleti Beosztás:")
                 st.dataframe(beosztas_df)
 
-                @st.cache_data
-                def convert_to_excel(data):
-                    import io
-                    output = io.BytesIO()
-                    with pd.ExcelWriter(output, engine="openpyxl") as writer:
-                        data.to_excel(writer, index=False)
-                    return output.getvalue()
-
-                excel_data = convert_to_excel(beosztas_df)
-                st.download_button(
-                    label="Beosztás letöltése Excelben",
-                    data=excel_data,
-                    file_name="ugyeleti_beosztas.xlsx",
-                    mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+                # Szűrési lehetőségek a felhasználónak
+                filter_options = st.multiselect(
+                    "Szűrd a generált ügyeleti beosztást az alábbi oszlopok alapján:",
+                    options=["Név", "Nap", "Indoklás"]
                 )
+
+                if filter_options:
+                    filtered_beosztas_df = beosztas_df[filter_options]
+                    st.write("Szűrt Ügyeleti Beosztás:")
+                    st.dataframe(filtered_beosztas_df)
+
+                    # Exportálás
+                    excel_data = convert_to_excel(filtered_beosztas_df)
+                    st.download_button(
+                        label="Szűrt beosztás letöltése Excelben",
+                        data=excel_data,
+                        file_name="szurt_ugyeleti_beosztas.xlsx",
+                        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+                    )
+                else:
+                    st.warning("Nem választottál ki oszlopokat a szűréshez.")
+
             else:
-                st.warning("Kérlek, válaszd ki az elérhetőségi napokat.")
+                st.warning("Kérlek, válaszd ki az elérhetőségi dátumokat.")
         else:
             st.warning("Nem található releváns adat a munkalapokon.")