From 35d73fe2eb92d161b44b3d976eae77a4086cd1fe Mon Sep 17 00:00:00 2001 From: Floke Date: Fri, 11 Apr 2025 11:38:57 +0000 Subject: [PATCH] bugfix --- brancheneinstufung.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/brancheneinstufung.py b/brancheneinstufung.py index ff18ccf5..4364c7f7 100644 --- a/brancheneinstufung.py +++ b/brancheneinstufung.py @@ -2286,6 +2286,10 @@ class DataProcessor: def main(): global MODE, LOG_FILE print("Bitte wählen Sie den Betriebsmodus:") + print("wiki: Nur Wikipedia-Verifizierung (Batch)") + print("website: Nur Website-Scraping (Batch)") + print("branch: Nur Brancheneinschätzung (Batch)") + print("combined: Alle Funktionen (Wikipedia, Website, Branch) in einem Durchlauf") print("1: Vollständige Verarbeitung (alle Funktionen)") print("11: Re-Evaluation markierter Zeilen (nur 'x' in Spalte A)") print("21: Website-Scraping Testmodus (nur Website-Rohtext & Zusammenfassung)") @@ -2293,13 +2297,12 @@ def main(): print("23: Website Detail Extraction (nur für Zeilen mit 'x')") print("31: Nur ChatGPT-Auswertung (alle ChatGPT-Routinen)") print("41: Nur Wikipedia-Scraping") - print("51: Batch-Verifizierung (alte Nummerierung beibehalten)") print("6: Contact Research (LinkedIn)") print("8: Batch Token-Zählung") - MODE = input("Geben Sie den Modus (Zahl) ein: ").strip() + MODE = input("Geben Sie den Modus ein (z.B. wiki, website, branch, combined oder alte Zahl): ").strip().lower() if not MODE: - MODE = "1" + MODE = "combined" LOG_FILE = create_log_filename(MODE) debug_print(f"Start Betriebsmodus {MODE}") @@ -2308,7 +2311,14 @@ def main(): dp = DataProcessor() - if MODE == "1": + if MODE in ["wiki", "website", "branch", "combined"]: + try: + row_limit = int(input("Wie viele Zeilen sollen insgesamt bearbeitet werden? ")) + except Exception as e: + debug_print(f"Fehler bei der Eingabe der Zeilenanzahl: {e}. Es werden alle Zeilen verarbeitet.") + row_limit = None + run_dispatcher(MODE, row_limit) + elif MODE == "1": dp.process_rows() elif MODE == "11": for i, row in enumerate(dp.sheet_handler.sheet_values[1:], start=2): @@ -2327,8 +2337,6 @@ def main(): elif MODE == "41": for i, row in enumerate(dp.sheet_handler.sheet_values[1:], start=2): dp._process_single_row(i, row, process_wiki=True, process_chatgpt=False) - elif MODE == "51": - process_verification_only() elif MODE == "6": process_contact_research() elif MODE == "8": @@ -2348,4 +2356,4 @@ def main(): print(f"Verarbeitung abgeschlossen. Logfile: {LOG_FILE}") if __name__ == '__main__': - main() \ No newline at end of file + main()