# import gradio as gr
# from translate import translate_text
# from image_generator import generate_image

# def chatbot(input_text, src_lang="auto"):
#     # Translate input to English
#     translated_text = translate_text(input_text, src_lang, "en")
#     # Generate image based on translated text
#     image_path = generate_image(f"A scene depicting: {translated_text}")
#     return translated_text, image_path

# interface = gr.Interface(
#     fn=chatbot,
#     inputs=[
#         gr.Textbox(label="Input Text"),
#         gr.Dropdown(choices=["auto", "es", "fr", "de"], label="Source Language")
#     ],
#     outputs=[
#         gr.Textbox(label="Translated Text"),
#         gr.Image(label="Generated Image")
#     ],
#     title="English Language Learning Chatbot"
# )

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


import gradio as gr
from translate import translate_text
from image_generator import generate_image

def chatbot(input_text, src_lang="auto"):
    """Process user input, translate it, and generate an image."""
    # Translate input to English
    translated_text = translate_text(input_text, src_lang, "en")
    
    # Generate an image based on the translated text
    image_path = generate_image(f"A scene depicting: {translated_text}")
    
    return translated_text, image_path

# Gradio Interface
interface = gr.Interface(
    fn=chatbot,
    inputs=[
        gr.Textbox(label="Enter text in any language"),
        gr.Textbox(label="Source Language (optional, e.g., 'es' for Spanish)", value="auto"),
    ],
    outputs=[
        gr.Textbox(label="Translated Text"),
        gr.Image(label="Generated Image"),
    ],
    title="LinguaVision - Multilingual Chatbot with Image Generation",
    description="Enter text in any language to translate it into English and generate an image based on the text.",
)

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


# import gradio as gr
# from translate import translate_text
# from image_generator import generate_image
# def chatbot(input_text, src_lang="auto"):
#     """Process user input, translate it, and generate an image."""
#     # Translate input to English
#     translated_text = translate_text(input_text, src_lang, "en")
#     # Generate an image based on the translated text
#     image_path = generate_image(f"A scene depicting: {translated_text}")
#     return translated_text, image_path
# # Gradio Interface
# interface = gr.Interface(
#     fn=chatbot,
#         inputs=[
# gr.Textbox(label="Enter text in any language"),
# gr.Textbox(label="Source Language (optional, e.g., 'es' for Spanish)", value="auto"),
#  ],
# outputs=[
# gr.Textbox(label="Translated Text"),
# gr.Image(label="Generated Image"),
#  ],
# title="LinguaVision - Multilingual Chatbot with Image Generation",
# description="Enter text in any language to translate it into English and generate an image based on the text.\n NOTE: This tool takes approximately 12 minutes to execute.",
# )
# if name == "main":
# interface.launch(share=True)