import os

import numpy as np

import gradio as gr
# Use a pipeline as a high-level helper
import requests

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 reverse_audio(text):
    data = query(text)
    return data#(16000, np.flipud(data))


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

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