bugfix

brancheneinstufung.py

@@ -921,35 +921,15 @@ def process_contacts():
 
 # ==================== MAIN PROGRAMM ====================
 if __name__ == "__main__":
-    print("Modi:")
+    import argparse
-    print("1 = Regulärer Modus")
+    parser = argparse.ArgumentParser()
-    print("2 = Re-Evaluierungsmodus (nur Zeilen mit 'x' in Spalte A)")
+    parser.add_argument("--mode", type=str, default="1", help="Modus: 1, 2, 3, 4, 5, 6 oder 7")
-    print("3 = Alignment-Demo (Header in Hauptblatt und Contacts)")
+    parser.add_argument("--num_rows", type=int, default=0, help="Anzahl der zu bearbeitenden Zeilen (nur für Modus 1)")
-    print("4 = Nur Wikipedia-Suche (Zeilen ohne Wikipedia-Timestamp)")
+    args = parser.parse_args()
-    print("5 = Nur ChatGPT-Bewertung (Zeilen ohne ChatGPT-Timestamp)")
+
-    print("6 = Contact Research (via SerpAPI)")
+    MODE = args.mode
-    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"
     if MODE == "1":
-        try:
+        num_rows = args.num_rows if args.num_rows > 0 else int(input("Wieviele Zeilen sollen überprüft werden? "))
-            num_rows = int(input("Wieviele Zeilen sollen überprüft werden? "))
-        except Exception as e:
-            print("Ungültige Eingabe. Bitte eine Zahl eingeben.")
-            exit(1)
         processor = DataProcessor()
         processor.process_rows(num_rows)
     elif MODE in ["2", "3"]: