Update app.py

app.py

@@ -23,7 +23,7 @@ if uploaded_file:
         # Betöltés Excel fájl
         excel_data = pd.ExcelFile(uploaded_file)
         sheet_names = excel_data.sheet_names
-        
+
         st.write("Munkalapok a fájlban:")
         st.write(sheet_names)
 
@@ -31,66 +31,80 @@ if uploaded_file:
         relevant_data = []
         for sheet in sheet_names:
             df = excel_data.parse(sheet)
-            if "Név" in df.columns and "Elérhetőség" in df.columns:
-                st.write(f"Adatok a '{sheet}' munkalapról:")
-                st.dataframe(df)
+            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.")
                 relevant_data.append(df)
-        
+            else:
+                st.write(f"A '{sheet}' munkalap nem tartalmaz releváns adatokat.")
+
         # Összesített adatok
         if relevant_data:
             combined_data = pd.concat(relevant_data, ignore_index=True)
             st.write("Kombinált adatok:")
             st.dataframe(combined_data)
 
-            # Adatfeldolgozás
-            combined_data["Elérhető napok"] = combined_data["Elérhetőség"].apply(lambda x: x.split(","))
-            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 range(1, 8):  # Példa: 7 napos beosztás
-                nap_nev = f"Nap {nap}"
-                for index, row in combined_data.iterrows():
-                    if nap_nev in row["Elérhető napok"]:
-                        korlatozott = any(
-                            foglalt_napok.get(nap_nev) == szemely
-                            for szemely in row["Korlátozások"]
-                        )
-                        if not korlatozott:
-                            foglalt_napok[nap_nev] = row["Név"]
-                            # Szöveg generálása
-                            prompt = f"{row['Név']} ügyel {nap_nev}-n, mert "
-                            indoklas = generator(prompt, max_length=50, num_return_sequences=1)[0]["generated_text"]
-                            beosztas.append({
-                                "Nap": nap_nev,
-                                "Orvos": row["Név"],
-                                "Indoklás": indoklas
-                            })
-                            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
-
-            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"
+            # Felhasználói input az elérhetőségi napokhoz
+            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"]
             )
 
+            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"]:
+                            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({
+                                    "Nap": nap,
+                                    "Orvos": row["Név"],
+                                    "Indoklás": indoklas
+                                })
+                                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
+
+                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"
+                )
+            else:
+                st.warning("Kérlek, válaszd ki az elérhetőségi napokat.")
         else:
             st.warning("Nem található releváns adat a munkalapokon.")