import os
import base64

import io
import requests
import numpy as np
from scipy.io import wavfile
from dotenv import load_dotenv


# Load environment variables
load_dotenv()


URL = os.getenv("URL")


# RAG API
def rag_api(query, audio_language_code):

    res = requests.post(f"{URL}/api/rag",
                        json={
                            'query': query,
                            'audio_language_code': audio_language_code
                        })

    return res.json()


# Agent + RAG API
def agent_api(query, audio_language_code):

    res = requests.post(f"{URL}/api/agent",
                        json={
                            'query': query,
                            'audio_language_code': audio_language_code
                        })

    return res.json()


def base64_to_audio(base64_string):

    try:
        # Decode the base64 string
        audio_data = base64.b64decode(base64_string)  # Decode
        audio_file = io.BytesIO(audio_data)  # Convert to BytesIO object

        sample_rate, samples = wavfile.read(audio_file)
        samples_array = np.array(samples)

        return sample_rate, samples_array
    except Exception as e:
        print(e)
        return None, None