Update app.py

app.py

@@ -11,7 +11,8 @@ st.write("Töltsd fel az orvosi adatokat tartalmazó Excel fájlt, és generálj
 # Hugging Face nyelvi modell betöltése
 @st.cache_resource
 def load_model():
-    return pipeline("text-generation", model="distilbert-base-multilingual-cased")
+    # Kompatibilis modell betöltése (pl. gpt2)
+    return pipeline("text-generation", model="gpt2")
 
 generator = load_model()
 
@@ -25,7 +26,7 @@ if uploaded_file:
         st.write("Feltöltött adatok:")
         st.dataframe(df)
 
-        # Előkészítés
+        # Adatok előkészítése
         df["Elérhető napok"] = df["Elérhetőség"].apply(lambda x: x.split(","))
         df["Korlátozások"] = df["Korlátozások"].apply(lambda x: x.split(",") if pd.notna(x) else [])
         foglalt_napok = {}
@@ -42,6 +43,7 @@ if uploaded_file:
                     )
                     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({
@@ -59,7 +61,12 @@ if uploaded_file:
         # Exportálás Excelbe
         @st.cache_data
         def convert_to_excel(data):
-            return data.to_excel(index=False, engine="openpyxl")
+            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(