Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import pandas as pd
 from transformers import pipeline
-from datetime import datetime
+from datetime import datetime, timedelta
+import plotly.express as px
 
 # Oldal konfiguráció
 st.set_page_config(page_title="Orvosi Ügyeleti Beosztás Generáló", layout="wide")
@@ -19,10 +20,8 @@ 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:
@@ -115,28 +114,29 @@ if uploaded_file:
                 st.write("Generált Ügyeleti Beosztás:")
                 st.dataframe(beosztas_df)
 
-                # 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"]
+                # Naptár nézet generálása Plotly segítségével
+                st.write("Naptár Nézet:")
+                beosztas_df["Nap"] = pd.to_datetime(beosztas_df["Nap"])
+                fig = px.timeline(
+                    beosztas_df,
+                    x_start="Nap",
+                    x_end="Nap",
+                    y="Orvos",
+                    title="Ügyeleti Beosztás",
+                    color="Orvos",
+                    labels={"Nap": "Dátum", "Orvos": "Orvos"}
+                )
+                fig.update_yaxes(categoryorder="total ascending")
+                st.plotly_chart(fig)
+
+                # Exportálás
+                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"
                 )
-
-                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 dátumokat.")
         else: