From 81ed87470e7e0aa8d3b6eb9a4b3bbbf9ed627711 Mon Sep 17 00:00:00 2001
From: Floke <floke.com@gmail.com>
Date: Thu, 3 Apr 2025 18:24:10 +0000
Subject: [PATCH] bugfix

---
 brancheneinstufung.py | 38 +++++++++-----------------------------
 1 file changed, 9 insertions(+), 29 deletions(-)

diff --git a/brancheneinstufung.py b/brancheneinstufung.py
index 1e9ed13b..6a01c0b4 100644
--- a/brancheneinstufung.py
+++ b/brancheneinstufung.py
@@ -921,35 +921,15 @@ def process_contacts():
 
 # ==================== MAIN PROGRAMM ====================
 if __name__ == "__main__":
-    print("Modi:")
-    print("1 = Regulärer Modus")
-    print("2 = Re-Evaluierungsmodus (nur Zeilen mit 'x' in Spalte A)")
-    print("3 = Alignment-Demo (Header in Hauptblatt und Contacts)")
-    print("4 = Nur Wikipedia-Suche (Zeilen ohne Wikipedia-Timestamp)")
-    print("5 = Nur ChatGPT-Bewertung (Zeilen ohne ChatGPT-Timestamp)")
-    print("6 = Contact Research (via SerpAPI)")
-    print("7 = Contacts (LinkedIn) – Kontakte in das Contacts-Blatt schreiben")
-    mode_input = input("Wählen Sie den Modus: ").strip()
-    if mode_input == "2":
-        MODE = "2"
-    elif mode_input == "3":
-        MODE = "3"
-    elif mode_input == "4":
-        MODE = "4"
-    elif mode_input == "5":
-        MODE = "5"
-    elif mode_input == "6":
-        MODE = "6"
-    elif mode_input == "7":
-        MODE = "7"
-    else:
-        MODE = "1"
+    import argparse
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--mode", type=str, default="1", help="Modus: 1, 2, 3, 4, 5, 6 oder 7")
+    parser.add_argument("--num_rows", type=int, default=0, help="Anzahl der zu bearbeitenden Zeilen (nur für Modus 1)")
+    args = parser.parse_args()
+
+    MODE = args.mode
     if MODE == "1":
-        try:
-            num_rows = int(input("Wieviele Zeilen sollen überprüft werden? "))
-        except Exception as e:
-            print("Ungültige Eingabe. Bitte eine Zahl eingeben.")
-            exit(1)
+        num_rows = args.num_rows if args.num_rows > 0 else int(input("Wieviele Zeilen sollen überprüft werden? "))
         processor = DataProcessor()
         processor.process_rows(num_rows)
     elif MODE in ["2", "3"]:
@@ -969,4 +949,4 @@ if __name__ == "__main__":
         process_contact_research()
     elif MODE == "7":
         process_contacts()
-    print(f"\n✅ Auswertung abgeschlossen ({Config.VERSION})")
+    print(f"\n✅ Auswertung abgeschlossen ({Config.VERSION})")
\ No newline at end of file