Update app.py

app.py

@@ -80,6 +80,17 @@ st.markdown(f"""
     </div>
 """, unsafe_allow_html=True)
 
+alle_heizsysteme = [
+    "Luft-Wasser Wärmepumpe",
+    "Wasser-Wasser Wärmepumpe",
+    "Sole-Wasser Wärmepumpe",
+    "Pelletheizung",
+    "Holzhackschnitzelheizung",
+    "Wasserstoffheizung",
+    "Gasheizung",
+    "Ölheizung",
+]
+
 if 'eingabedaten' not in st.session_state:
     st.session_state.eingabedaten = None
 if "df_heizsysteme" not in st.session_state:
@@ -661,16 +672,24 @@ elif modus == "Upload csv-Datei":
                                                                  zinssatz, beobachtungszeitraum), axis=1)
                             df["Annuität_NS"] = 0
                             df["Annuität"] = df["Annuität_NK"] + df["Annuität_NV"] + df["Annuität_NB"] + df["Annuität_NS"]
-                            df = df.sort_values("Annuität")
-                            df_anni = df[[
-                                "Name", "Annuität_NK", "Annuität_NV", "Annuität_NB", "Annuität"
-                            ]].copy()
-                            df_anni["Objekt-ID"] = row.get("Objekt-ID", idx)
-                            annuitaeten_gesamt.append(df_anni)
-
-                            for j in range(min(len(df), max_count)):
-                                df_out.at[idx, annuität_colnames[j]] = int(df.iloc[j]["Annuität"])
-                                df_out.at[idx, hsystem_colnames[j]] = df.iloc[j]["Name"]
+
+                            ann_dict = dict(zip(df["Name"], df["Annuität"]))
+                            günstigstes_sys = min(ann_dict.items(), key=lambda x: x[1])[0] if ann_dict else ""
+                            df_out.at[idx, "Günstigste Alternative"] = günstigstes_sys
+                            for sys in alle_heizsysteme:
+                                wert = ann_dict.get(sys, None)
+                                df_out.at[idx, sys] = wert
+                                
+                            # df = df.sort_values("Annuität")
+                            # df_anni = df[[
+                            #     "Name", "Annuität_NK", "Annuität_NV", "Annuität_NB", "Annuität"
+                            # ]].copy()
+                            # df_anni["Objekt-ID"] = row.get("Objekt-ID", idx)
+                            # annuitaeten_gesamt.append(df_anni)
+
+                            # for j in range(min(len(df), max_count)):
+                            #     df_out.at[idx, annuität_colnames[j]] = int(df.iloc[j]["Annuität"])
+                            #     df_out.at[idx, hsystem_colnames[j]] = df.iloc[j]["Name"]
                         except Exception as e:
                             st.warning(f"Objekt-ID {row.get('Objekt-ID', idx)}: {e}")