import streamlit as st  # Make sure to import streamlit
import torch
from transformers import MarianMTModel, MarianTokenizer
import speech_recognition as sr
from gtts import gTTS
import os
import datetime

def is_allowed_time():
    current_time = datetime.datetime.now().time()
    if current_time >= datetime.time(18, 0):  # After 6 PM IST
        return True
    else:
        return False

st.title("English to Hindi Audio Translation")

if is_allowed_time():
    audio_file = st.file_uploader("Upload an audio file", type=["wav", "mp3"])
    if audio_file is not None:
        text = recognize_speech_from_file(audio_file)
        st.write(f"Recognized English Text: {text}")
        translated_text = translate_text(text)
        st.write(f"Translated Hindi Text: {translated_text}")
        audio_output = convert_to_audio(translated_text)
        st.audio(audio_output, format='audio/mp3')
else:
    st.write("Translation service is available after 6 PM IST.")