from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.firefox.options import Options
import time
import os

# Verzeichnisse
input_directory = "actors_split"
download_directory = "actors_split_translated"
os.makedirs(download_directory, exist_ok=True)

# Geckodriver-Pfad (anpassen)
geckodriver_path = "/Pfad/zu/deinem/geckodriver"

# Firefox-Optionen
options = Options()
options.headless = False  # Setze auf True, wenn du den Browser nicht sehen willst

# Webdriver starten
driver = webdriver.Firefox(executable_path=geckodriver_path, options=options)

# ChatGPT-Website öffnen
driver.get("https://chat.openai.com/")

# Warten, bis du dich eingeloggt hast
input("Bitte logge dich in ChatGPT ein und drücke dann Enter...")

# Dateien hochladen
for file_name in os.listdir(input_directory):
    if file_name.endswith(".json"):
        file_path = os.path.abspath(os.path.join(input_directory, file_name))
        
        # Datei-Upload starten
        upload_button = driver.find_element(By.XPATH, "//input[@type='file']")
        upload_button.send_keys(file_path)
        print(f"Datei hochgeladen: {file_name}")
        
        # Warte auf die Antwort und download
        time.sleep(20)  # Wartezeit anpassen je nach Dateigröße
        
        # Speichern der übersetzten Datei
        with open(os.path.join(download_directory, file_name), 'w', encoding='utf-8') as out_file:
            out_file.write(driver.page_source)
        
        print(f"Übersetzte Datei gespeichert: {file_name}")

# Browser schließen
driver.quit()