changes to how the webgui performs on hugging face

modules/common/file_ensurer.py

@@ -64,6 +64,9 @@ class FileEnsurer():
     openai_api_key_path = os.path.join(secrets_dir,'openai_api_key.txt')
     gemini_api_key_path = os.path.join(secrets_dir,'gemini_api_key.txt')
     google_translate_service_key_json_path = os.path.join(secrets_dir, "google_translate_service_key.json")
+
+    ## temp files
+    temp_file_path = os.path.join(script_dir, "temp.txt")
 
     ## favicon
     favicon_path = os.path.join(gui_lib, "Kudasai_Logo.png")
@@ -219,9 +222,7 @@ def purge_storage() -> None:
         logging.debug("Running on Hugging Face, purging storage")
 
         stuff_to_purge = [
-            FileEnsurer.secrets_dir,
-            FileEnsurer.config_dir,
-            FileEnsurer.archive_dir
+            FileEnsurer.temp_file_path
         ]
 
         stuff_to_truncate = [
@@ -453,6 +454,9 @@ def archive_results(list_of_result_tuples:typing.List[typing.Tuple[str,str]], mo
 
         """
 
+        if(FileEnsurer.is_hugging_space()):
+            return
+
         archival_path = os.path.join(FileEnsurer.archive_dir, f'{module}_run_{timestamp}')
         FileEnsurer.standard_create_directory(archival_path)
 

webgui.py

@@ -2,6 +2,7 @@
 import typing
 import base64
 import asyncio
+import os
 
 ## third-party libraries
 import gradio as gr
@@ -137,6 +138,9 @@ def get_saved_api_key(service_name:typing.Literal["openai","gemini","deepl","goo
 
                 """
 
+                if(FileEnsurer.is_hugging_space()):
+                    return ""
+
                 service_to_path = {
                     "openai": FileEnsurer.openai_api_key_path,
                     "gemini": FileEnsurer.gemini_api_key_path,
@@ -169,10 +173,14 @@ async def set_translator_api_key(api_key) -> None:
 
                 """
 
+                ## ik this looks really bad
+                ## but the set_credentials for google translate expects a filepath, and normal way isn't working
+                ## so we write the client secrets to a temp file, send that, then clear the file
+                ## even if it fails, purge_storage is set to kill that path too so we should be good.
                 if(Translator.TRANSLATION_METHOD == "google translate"):
-                    FileEnsurer.standard_overwrite_file(FileEnsurer.google_translate_service_key_json_path, str(api_key), omit=True)
-                    await asyncio.sleep(2)
-                    api_key = FileEnsurer.google_translate_service_key_json_path
+                    FileEnsurer.standard_overwrite_file(FileEnsurer.temp_file_path, str(api_key), omit=True)
+                    await asyncio.sleep(1)
+                    api_key = FileEnsurer.temp_file_path
 
                 try:
 
@@ -186,6 +194,13 @@ async def set_translator_api_key(api_key) -> None:
                 except:
                     raise gr.Error("Invalid API key")
 
+                finally:
+                    try:
+                        os.remove(FileEnsurer.temp_file_path)
+
+                    except:
+                        pass
+
 ##-------------------start-of-update_translator_api_key()---------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 
             def update_translator_api_key(api_key) -> None:
@@ -199,6 +214,9 @@ def update_translator_api_key(api_key) -> None:
 
                 """
 
+                if(FileEnsurer.is_hugging_space()):
+                    return
+
                 method_to_path = {
                     "openai": FileEnsurer.openai_api_key_path,
                     "gemini": FileEnsurer.gemini_api_key_path,