Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 from transformers import pipeline
+from datetime import datetime
 
 # Oldal konfiguráció
 st.set_page_config(page_title="Orvosi Ügyeleti Beosztás Generáló", layout="wide")
@@ -15,6 +16,14 @@ def load_model():
 
 generator = load_model()
 
+# Függvény a dátumok felismerésére és konverziójára
+def parse_date(value):
+    try:
+        parsed_date = datetime.strptime(value, "%d %B")
+        return parsed_date.replace(year=2025)
+    except ValueError:
+        return None
+
 # Fájl feltöltése
 uploaded_file = st.file_uploader("Tölts fel egy Excel fájlt", type=["xlsx"])
 
@@ -34,7 +43,6 @@ if uploaded_file:
             st.write(f"Elemzés a '{sheet}' munkalapon:")
             st.dataframe(df.head())
 
-            # Oszlopok azonosítása
             oszlopok = df.columns.str.lower()
             if any("név" in col for col in oszlopok):
                 st.write(f"Releváns adat található a '{sheet}' munkalapon.")
@@ -48,30 +56,32 @@ if uploaded_file:
             st.write("Kombinált adatok:")
             st.dataframe(combined_data)
 
-            # Felhasználói input az elérhetőségi napokhoz
+            # 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 x.split(",")] if isinstance(x, str) else []
+            )
+            combined_data["Korlátozások"] = combined_data["Korlátozások"].apply(
+                lambda x: x.split(",") if isinstance(x, str) else []
+            )
+
             available_days = st.multiselect(
-                "Add meg az elérhetőségi napokat (pl. Hétfő, Kedd, ...):",
-                options=["Hétfő", "Kedd", "Szerda", "Csütörtök", "Péntek", "Szombat", "Vasárnap"]
+                "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]
             )
 
             if available_days:
-                combined_data["Elérhető napok"] = [available_days] * len(combined_data)
-
-                # Adatfeldolgozás
-                combined_data["Korlátozások"] = combined_data["Korlátozások"].apply(lambda x: x.split(",") if pd.notna(x) else [])
                 foglalt_napok = {}
                 beosztas = []
 
                 for nap in available_days:
                     for index, row in combined_data.iterrows():
-                        if nap in row["Elérhető napok"]:
+                        if nap in [day.strftime("%Y-%m-%d") for day in row["Elérhető napok"] if day]:
                             korlatozott = any(
                                 foglalt_napok.get(nap) == szemely
                                 for szemely in row["Korlátozások"]
                             )
                             if not korlatozott:
                                 foglalt_napok[nap] = row["Név"]
-                                # Szöveg generálása
                                 prompt = f"{row['Név']} ügyel {nap}-n, mert "
                                 indoklas = generator(prompt, max_length=50, num_return_sequences=1)[0]["generated_text"]
                                 beosztas.append({
@@ -81,20 +91,17 @@ 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)
 
-                # Exportálás Excelbe
                 @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)
-                    processed_data = output.getvalue()
-                    return processed_data
+                    return output.getvalue()
 
                 excel_data = convert_to_excel(beosztas_df)
                 st.download_button(