import gradio as gr
import requests
import os

# Hugging Face API Key from environment variables
HF_API_KEY = os.getenv('HF_API_KEY')

# Function to fetch word details from Hugging Face API
def fetch_word_details(word):
    if not word:
        return {"error": "Word is required"}
    
    api_url = "https://api-inference.huggingface.co/models/gauravkai/Llama-3-8B-English-Dictionary"
    prompt = f"Provide the following details for the word '{word}': \n    - Definition\n    - Syllables\n    - Phonetic Transcription\n    - Example Sentence\n    - Phonemes\n    Please return the result in a structured JSON"

    headers = {
        "Authorization": f"Bearer {HF_API_KEY}",
        "Content-Type": "application/json",
    }

    # Sending the request to Hugging Face API
    response = requests.post(api_url, headers=headers, json={"inputs": prompt})

    if response.status_code == 200:
        result = response.json()
        word_details = parse_details(result[0]['generated_text'])
        return word_details
    else:
        return {"error": "Failed to fetch data"}

def parse_details(text):
    # Extract the JSON part of the string
    json_start = text.find("{")
    json_end = text.rfind("}")
    if json_start != -1 and json_end != -1:
        json_string = text[json_start:json_end+1]

        # Fix unquoted values for JSON compatibility
        json_string = json_string.replace(r'\/', "/")

        try:
            return eval(json_string)  # Safely parse the string to a dictionary
        except Exception as e:
            return {"error": f"Failed to parse JSON: {str(e)}"}
    return {"error": "No valid JSON found."}

# Gradio Interface
def word_details_interface(word):
    details = fetch_word_details(word)
    if "error" in details:
        return details["error"]
    
    return f"""
        <h3>Word Details</h3>
        <b>Word:</b> {details.get('word', 'N/A')}<br>
        <b>Definition:</b> {details.get('definition', 'N/A')}<br>
        <b>Syllables:</b> {details.get('syllables', 'N/A')}<br>
        <b>Phonetic Transcription:</b> {details.get('phonetic_transcription', 'N/A')}<br>
        <b>Example Sentence:</b> {details.get('example_sentence', 'N/A')}<br>
        <b>Phonemes:</b> {details.get('phonemes', 'N/A')}<br>
    """

# Gradio UI Setup
interface = gr.Interface(
    fn=word_details_interface,
    inputs=gr.Textbox(label="Enter a Word"),
    outputs=gr.HTML(),
    live=True,
)

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