Update app.py

app.py

@@ -21,7 +21,7 @@ import easyocr
 import tts 
 
 
-
+gpt_state = 0
 
 article = """
 <div style='margin:20px auto;'>
@@ -112,12 +112,15 @@ def init_openai_api_key(api_key=""):
     print(text_refiner)
     openai_available = text_refiner is not None
     if openai_available:
-        return [gr.update(visible=True)]+[gr.update(visible=False)]+[gr.update(visible=True)]*3+[gr.update(visible=False)]+ [gr.update(visible=True)]+ [gr.update(visible=False)]*2 + [text_refiner, visual_chatgpt, None]+[1]
+
+        global gpt_state
+        gpt_state=1
+        return [gr.update(visible=True)]+[gr.update(visible=False)]+[gr.update(visible=True)]*3+[gr.update(visible=False)]+ [gr.update(visible=True)]+ [gr.update(visible=False)]*2 + [text_refiner, visual_chatgpt, None]
     else:
         return [gr.update(visible=False)]*7 + [gr.update(visible=True)]*2 + [text_refiner, visual_chatgpt, 'Your OpenAI API Key is not available']
         
 def init_wo_openai_api_key():
-        return  [gr.update(visible=False)]*4 + [gr.update(visible=True)]+ [gr.update(visible=False)]+[gr.update(visible=True)]+[gr.update(visible=False)]*2 + [None, None, None]+[0]
+        return  [gr.update(visible=False)]*4 + [gr.update(visible=True)]+ [gr.update(visible=False)]+[gr.update(visible=True)]+[gr.update(visible=False)]*2 + [None, None, None]
 
 def get_click_prompt(chat_input, click_state, click_mode):
     inputs = json.loads(chat_input)
@@ -494,7 +497,7 @@ def create_ui():
         input_mask_state = gr.State(np.zeros((1, 1)))
         input_points_state = gr.State([])
         input_labels_state = gr.State([])
-        gpt_state = gr.State(0)
+
 
         gr.Markdown(title)
         gr.Markdown(description)
@@ -611,7 +614,7 @@ def create_ui():
                     use_mic = gr.Checkbox(label="Check to use Microphone as Reference", value=False)
                     agree = gr.Checkbox(label="Agree", value=True)
                     output_waveform = gr.Video(label="Waveform Visual")
-                    output_audio = gr.Audio(label="Synthesised Audio",autoplay=True)
+                    output_audio = gr.HTML(label="Synthesised Audio")
 
                     with gr.Row():
                         submit_tts = gr.Button(value="Submit", interactive=True)
@@ -638,11 +641,11 @@ def create_ui():
         
         openai_api_key.submit(init_openai_api_key, inputs=[openai_api_key],
                               outputs=[modules_need_gpt0, modules_need_gpt1, modules_need_gpt2, modules_need_gpt3, modules_not_need_gpt,
-                                       modules_not_need_gpt2, tts_interface,module_key_input ,module_notification_box, text_refiner, visual_chatgpt, notification_box,gpt_state])
+                                       modules_not_need_gpt2, tts_interface,module_key_input ,module_notification_box, text_refiner, visual_chatgpt, notification_box])
         enable_chatGPT_button.click(init_openai_api_key, inputs=[openai_api_key],
                                     outputs=[modules_need_gpt0, modules_need_gpt1, modules_need_gpt2, modules_need_gpt3,
                                              modules_not_need_gpt,
-                                             modules_not_need_gpt2, tts_interface,module_key_input,module_notification_box, text_refiner, visual_chatgpt, notification_box,gpt_state])
+                                             modules_not_need_gpt2, tts_interface,module_key_input,module_notification_box, text_refiner, visual_chatgpt, notification_box])
         disable_chatGPT_button.click(init_wo_openai_api_key,
                                      outputs=[modules_need_gpt0, modules_need_gpt1, modules_need_gpt2, modules_need_gpt3,
                                               modules_not_need_gpt,
@@ -733,6 +736,7 @@ def create_ui():
                 return [gr.update(visible=True)]*2+[gr.update(visible=False)]*2
             else: 
                 print("no gpt")
+                print("gpt_state",gpt_state)
                 return [gr.update(visible=False)]+[gr.update(visible=True)]+[gr.update(visible=False)]*2
         
         def on_base_selected():
@@ -790,12 +794,8 @@ def create_ui():
             show_progress=False, queue=True
         )
 
-        # def update_output_audio():
-        #     return gr.update(autoplay=True)
-        
-        output_audio.change(update_output_audio,outputs=[output_audio])
 
-       
+
 
 
         return iface
@@ -804,4 +804,4 @@ def create_ui():
 if __name__ == '__main__':
     iface = create_ui()
     iface.queue(concurrency_count=5, api_open=False, max_size=10)
-    iface.launch(server_name="0.0.0.0", enable_queue=True)
+    iface.launch(server_name="0.0.0.0", enable_queue=True)