import streamlit as st
import google.generativeai as genai
import os
from dotenv import load_dotenv
import requests
load_dotenv()

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

model = genai.GenerativeModel("gemini-1.5-pro-latest")
prompt = """
Input Text: {tamil_text} (Tanglish)

Task:

1. Translate the input text to English.
2. Identify and explain grammatical errors.
3. Explain word choice and provide synonyms.

give me the output in this format:

1. English Translation: The user input translated into grammatically correct English.
2. Grammar Explanation: A clear explanation of any grammatical errors in the user input (Tanglish) and suggestions for correction in Tamil.
3. Word Choice Explanation and Synonyms: Explanation of vocabulary meaning and synonyms in both English and Tamil.
"""

prompt_1 = """
You are an English teacher. Please understand the following Tanglish query and provide a correct English sentence with correct grammar:
input:
Tanglish query: {tamil_text}

example input 1: epdi amma kita pasikidhu nu english la soldradhu
example Output 1: mom, i am hungry

example input 2: teacher kita bathroom poonu tu epdi kekuradhu
example Output 2: Excuse me, teacher. May I please be excused to use the restroom?

note output must in one line
"""

def generate_response(input_text, prompt):
  query = prompt.format(tamil_text=input_text)
  response = model.generate_content(query)
  response = response.text
  return st.markdown(response)

def txt2speech(text):
  API_URL = "https://api-inference.huggingface.co/models/espnet/kan-bayashi_ljspeech_vits"
  api_token = os.getenv('HUGGING_FACE')
  headers = {"Authorization": f"Bearer {api_token}"}
  payloads = {'inputs': text}

  response = requests.post(API_URL, headers=headers, json=payloads)
  
  with open('audio_answer.mp3', 'wb') as file:
      file.write(response.content)
  

st.title("English Teaching AI")
user_query = st.text_area("Type Tamil or Tanglish sentance")
submit = st.button("Analyze")
if submit:
    with st.spinner("Processing..."):
        answer = generate_response(user_query, prompt_1)
        txt2speech(answer)
        st.audio("audio_answer.mp3")
    with st.spinner("Analyzing..."):
        generate_response(user_query, prompt)