app.py

import os

import numpy as np

import gradio as gr
import requests
from genai_chat_ai import AI,create_chat_session
API_URL = "https://api-inference.huggingface.co/models/asg2024/vits-ar-sa"
api_key = os.environ.get("Id_mode_vits") 
headers = {"Authorization": f"Bearer {api_key}"}

def query(text):
  payload={"inputs": text}
  response = requests.post(API_URL, headers=headers, json=payload)
  return response.content

def   get_answer_ai(text):
      global AI
      try:
          response = AI.send_message(text)
          return response.text

          
      except error:
          AI=create_chat_session()
          response = AI.send_message(text)
          return response.text
          
def reverse_audio(text):
    text_answer=get_answer_ai(text)
    
    data = query(text_answer)
    return data,text_answer#(16000, np.flipud(data))


demo = gr.Interface(
    fn=reverse_audio,
    inputs="text",
    outputs=["audio","text"]
)

if __name__ == "__main__":
    demo.launch()