#!/usr/bin/env python

# WS server example

import asyncio
import websockets

# Imports the Google Cloud client library
from google.cloud import speech

# Instantiates a client
client = speech.SpeechClient()


# config = speech.RecognitionConfig(
#     encoding=speech.RecognitionConfig.AudioEncoding.LINEAR16,
#     sample_rate_hertz=16000,
#     language_code="en-US",
# )

# streaming_config = speech.StreamingRecognitionConfig(config=config)

# def transcribe(stream):

#     requests = (
#         speech.StreamingRecognizeRequest(audio_content=chunk) for chunk in stream
#     )

#     # Detects speech in the audio file
#     responses = client.streaming_recognize(
#             config=config,
#             requests=requests
#     )

#     for response in responses:
#             # Once the transcription has settled, the first result will contain the
#             # is_final result. The other results will be for subsequent portions of
#             # the audio.
#             for result in response.results:
#                 print("Finished: {}".format(result.is_final))
#                 print("Stability: {}".format(result.stability))
#                 alternatives = result.alternatives
#                 # The alternatives are ordered from most likely to least.
#                 for alternative in alternatives:
#                     print("Confidence: {}".format(alternative.confidence))
#                     print(u"Transcript: {}".format(alternative.transcript))


speechtotext_client = speech.SpeechClient()

config = speech.RecognitionConfig(
    audio_channel_count=1,
    encoding=speech.RecognitionConfig.AudioEncoding.LINEAR16,
    sample_rate_hertz=44100,
    language_code="en-US",
)

# Transcribe an audio file.
def transcribe(content):
    input(content)
    input(type(content))

    content = speech.RecognitionAudio(content=content)

    # Detects speech in the audio file
    response = speechtotext_client.recognize(config=config, audio=content)
    input(f"Response: {response}")

    for result in response.results:
        return result.alternatives[0].transcript


async def session(websocket, path):
    buf = []

    while True:
        chunk = await websocket.recv()

        if chunk == "stop":
            break

        buf.append(chunk)

    input(f"Buffer sample: {buf[0]}")
    audio = b"".join(buf)

    transcribe(audio)


start_server = websockets.serve(session, "localhost", 8080)
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()