Update app.py

app.py

@@ -1,8 +1,7 @@
 import streamlit as st
 import pandas as pd
 from transformers import pipeline
-from datetime import datetime, timedelta
-import plotly.express as px
+from datetime import datetime
 
 # Oldal konfiguráció
 st.set_page_config(page_title="Orvosi Ügyeleti Beosztás Generáló", layout="wide")
@@ -24,9 +23,16 @@ def parse_date(value):
             return datetime.strptime(value, "%d %B %Y")
         parsed_date = datetime.strptime(value, "%d %B")
         return parsed_date.replace(year=2025)
-    except ValueError:
+    except (ValueError, TypeError):
         return None
 
+# Adatok tisztítása és egységesítése
+def clean_data(df):
+    # Oszlopok egységesítése
+    df.columns = df.columns.str.strip().str.lower()
+    df = df.rename(columns={"név": "Név", "elérhetőség": "Elérhetőség", "korlátozások": "Korlátozások"})
+    return df
+
 # Adatok Excel exportálása
 @st.cache_data
 def convert_to_excel(data):
@@ -52,20 +58,14 @@ if uploaded_file:
         relevant_data = []
         for sheet in sheet_names:
             df = excel_data.parse(sheet)
-            st.write(f"Elemzés a '{sheet}' munkalapon:")
-            st.dataframe(df.head())
+            df = clean_data(df)  # Adatok tisztítása és egységesítése
 
-            oszlopok = df.columns.str.lower()
-            if any("név" in col for col in oszlopok):
+            # Ellenőrzés, hogy tartalmaz-e releváns oszlopokat
+            if "Név" in df.columns and "Elérhetőség" in df.columns:
                 st.write(f"Releváns adat található a '{sheet}' munkalapon.")
                 relevant_data.append(df)
             else:
-                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)
+                st.warning(f"A '{sheet}' munkalap nem tartalmaz releváns adatokat.")
 
         # Összesített adatok
         if relevant_data:
@@ -114,21 +114,6 @@ if uploaded_file:
                 st.write("Generált Ügyeleti Beosztás:")
                 st.dataframe(beosztas_df)
 
-                # 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(