Update app.py

app.py

@@ -166,17 +166,17 @@ def generate_audio(text,speaker_id=None):
             yield  16000,chunk#.squeeze().cpu().numpy()#.astype(np.int16).tobytes() 
 
 
-def   get_answer_ai(text):
-      global AI
+def   get_answer_ai(text,session_ai):
+     
       try:
-          response = AI.send_message(text,stream=True)
+          response = session_ai.send_message(text,stream=True)
           return response
 
           
       except :
-          AI=create_chat_session()
-          response = AI.send_message(text,stream=True)
-          return response
+          session_ai=create_chat_session()
+          response = session_ai.send_message(text,stream=True)
+          return response,session_ai
 
 def   modelspeech(text):
      with torch.no_grad():
@@ -194,29 +194,29 @@ def clean_text(text):
   return cleaned_text.strip()  # Remove leading/trailing spaces
 
 
-def text_to_speech(text):
+def text_to_speech(text,session_ai):
     
-    response = dash(text,False)
+    response = dash(text,session_ai,False)
     pad_text=''
     k=0
     for chunk in response:
-       
+       chunk,session_ai=chunk
        pad_text+=str(clean_text(chunk))
        
        if pad_text!='' and len(pad_text)>10:
            out=pad_text
            pad_text=''
            k+=1
-           yield modelspeech(out)
+           yield modelspeech(out),session_ai
            # for   stream_wav in generate_audio(out):
            #      yield stream_wav
     if pad_text!='':
-        yield modelspeech(pad_text)
+        yield modelspeech(pad_text),session_ai
        # for   stream_wav in generate_audio(pad_text):
        #          yield stream_wav
-def dash(text,is_state=True):
+def dash(text,session_ai,is_state=True):
     
-    response=get_answer_ai(text)
+    response,session_ai=get_answer_ai(text,session_ai)
     txt=' '
     for chunk in  response:
         if is_state:
@@ -225,7 +225,7 @@ def dash(text,is_state=True):
            txt=chunk.text
         
         
-        yield txt
+        yield txt,session_ai
 
 
 
@@ -234,17 +234,18 @@ def dash(text,is_state=True):
 # demo.launch()
 
 with gr.Blocks() as demo:
+    session_ai=gr.State()
     with gr.Tab("AI Text  "):
         gr.Markdown("# Text to Speech")
         text_input = gr.Textbox(label="Enter Text")
         text_out = gr.Textbox()
         
-        text_input.submit(dash, [text_input], text_out)
+        text_input.submit(dash, [text_input,session_ai],[text_out,session_ai])
     with gr.Tab("AI Speech"):
         gr.Markdown("# Text to Speech")
         text_input2 = gr.Textbox(label="Enter Text")
         audio_output = gr.Audio(streaming=True,autoplay=True)
-        text_input2.submit(text_to_speech, text_input2, audio_output)
+        text_input2.submit(text_to_speech, [text_input2,session_ai], [audio_output,session_ai])
          
 
 demo.launch(show_error=True)