add mac mps

app.py

@@ -1,4 +1,6 @@
 import os
+import time
+import platform
 
 def is_running_in_huggingface_spaces():
     return "SPACE_ID" in os.environ
@@ -24,7 +26,13 @@ try:
     lang=config.lang
 except ImportError:
     is_config_ui_installed=False
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    if platform.system() == "Darwin" and platform.machine().startswith("arm"):
+        print("run on mac with Apple Silicon")
+        if torch.backends.mps.is_available():
+            device = torch.device("mps")  # MPS = Metal Performance Shaders
+    else:
+        #TODO parse env var to assign cuda device
+        device = 0 if torch.cuda.is_available() else "cpu"
     lang='EN'
 
 try:
@@ -42,6 +50,9 @@ max_target_length = 256
 prefix = "enhance prompt"
 
 def enhance_prompt(prompt, system_prompt, temperature=0.5, repetition_penalty=1.2, seed=-1, is_rnd_seed=True):
+    global lang
+    start_time = time.time()  # Начало замера времени
+    
 
     if is_rnd_seed or seed==-1:
         seed = torch.randint(0, 2**32 - 1, (1,)).item()
@@ -73,8 +84,12 @@ def enhance_prompt(prompt, system_prompt, temperature=0.5, repetition_penalty=1.
         result_output_ru = translator.translate_eng2ru(generated_text_en)
     else:
         result_output_ru=generated_text_en
+
+    end_time = time.time()  # Конец замера времени
+    execution_time = end_time - start_time
+    time_str=f"execution time: {execution_time:.2f} s." if lang=="EN" else f"время выполнения: {execution_time:.2f} с."
     
-    return seed, generated_text_en, result_output_ru
+    return seed, generated_text_en, result_output_ru, time_str
 
 def random_prompt():
     rnd_prompt_str=generate_random_portrait_prompt()
@@ -176,8 +191,8 @@ def set_initial():
         device_name = f"GPU: {device_name}"
         dev="cuda"
     else:
-        device_name = "use CPU"
-    return gr.update(value=lang), gr.update(value=dev), f'{device_name}\nset to "{lang}" language'
+        device_name = "running on CPU"
+    return gr.update(value=lang), gr.update(value=dev), f'{device_name}, set to "{lang}" language'
 
 
 # Настройка интерфейса Gradio
@@ -196,8 +211,14 @@ with gr.Blocks(title="Flux Prompt Enhance",
             radio_lang = gr.Radio(choices = ["RU",  "EN"],  show_label = False, container = False, type = "value",
                                 visible = True if is_google_translate_installed else False)
 
-            radio_gpu  = gr.Radio(choices = ["cuda","cpu"], show_label = False, container = False, type = "value",
-                                visible = True if torch.cuda.is_available() else False)
+            if platform.system() == "Darwin" and platform.machine().startswith("arm") and torch.backends.mps.is_available():
+                print("radio_gpu mac")
+                radio_gpu  = gr.Radio(choices = ["mps","cpu"], show_label = False, container = False, type = "value",
+                                    visible = True)
+            else:
+                radio_gpu  = gr.Radio(choices = ["cuda","cpu"], show_label = False, container = False, type = "value",
+                                    visible = True if torch.cuda.is_available() else False)
+            
             save_button = gr.Button(LABELS["save_button"], visible= True if is_config_ui_installed else False)
         
     with gr.Row(variant="default"):
@@ -227,28 +248,44 @@ with gr.Blocks(title="Flux Prompt Enhance",
     # Кнопка генерации
     with gr.Row(variant="default"):
         generate_button = gr.Button(LABELS["generate_button"], variant="primary", size="lg")
-        generate_button.click(fn=enhance_prompt, inputs=[prompt_input,system_prompt,temperature,repetition_penalty,seed_output,is_rnd_seed],
-                              outputs=[seed_output, result_output, result_output_ru])
     
     # Кнопка копирования в буфер обмена
         copy_button = gr.Button(LABELS["copy_button"], variant="secondary")
         copy_button.click(fn=copy_to_clipboard, inputs=result_output, outputs=[],js="(text) => navigator.clipboard.writeText(text)")
+        
     with gr.Row(variant="default"):
-        log_text = gr.Textbox(label="")
+        log_text = gr.Textbox(label="", container=False)
     
     if is_config_ui_installed:    
         save_button.click(fn=save_config, inputs=[], outputs=log_text)
 
-    #preload values for lang
-    demo.load(set_initial, outputs=[radio_lang, radio_gpu, log_text])
-    
+    generate_button.click(fn=enhance_prompt, inputs=[prompt_input,system_prompt,temperature,repetition_penalty,seed_output,is_rnd_seed],
+                              outputs=[seed_output, result_output, result_output_ru,log_text])
+  
     if is_google_translate_installed:
         radio_lang.change(process_lang, inputs=radio_lang,
                         outputs=[log_text,generate_button, copy_button, save_button, prompt_input, seed_output, is_rnd_seed,
                                 result_output, result_output_ru,AccordionAdvanced,system_prompt, temperature, repetition_penalty])
 
     radio_gpu.change(process_gpu, inputs=radio_gpu, outputs=log_text)
+    
+    #preload values for lang
+    demo.load(set_initial, outputs=[radio_lang, radio_gpu, log_text])
+
+
+launch_args={}
+
+if not is_running_in_huggingface_spaces():
+    launch_args["share"]=False
+    launch_args["server_name"]="0.0.0.0"
+    launch_args["inbrowser"] = True
+    launch_args["favicon_path"] = "./static/favicon_aicave.png"
+    launch_args["show_api"]=True
+    
+    if os.path.exists("cert.pem") and os.path.exists("key.pem"):
+        launch_args["ssl_certfile"]="cert.pem"
+        launch_args["ssl_keyfile"]="key.pem"
+        launch_args["ssl_verify"]=False
 
 
-# Запуск приложения с прослушиванием на всех интерфейсах и открытием в браузере
-demo.launch()
+demo.launch(**launch_args)