Spaces:
Sleeping
Sleeping
| # from gpt_module import generate_response | |
| # from dalle_module import generate_image | |
| # from translation_module import translation_to_english | |
| # from tts_module import text_to_speech | |
| # from stt import transcribe_audio | |
| # def main(): | |
| # print("Welcome to the ChatGPT CLI application!") | |
| # while True: | |
| # user_input = input("Enter command: ") | |
| # if user_input.startswith('/imagine'): | |
| # img = user_input.replace('/imagine', '').strip() | |
| # print(generate_image(img)) | |
| # elif user_input.startswith('/chat'): | |
| # prompt = user_input.replace('/chat', '').strip() | |
| # print(generate_response(prompt)) | |
| # elif user_input.startswith('/translate'): | |
| # tra = user_input.replace('/translate', '').strip() | |
| # print(translation_to_english(tra)) | |
| # elif user_input.startswith('/voice'): | |
| # text = user_input.replace('/voice', '').strip() | |
| # print(text_to_speech(text)) | |
| # elif user_input.startswith('/transcribe'): | |
| # print("Please upload the audio file.") | |
| # audio_file = input("Enter the path to the audio file: ") | |
| # print(transcribe_audio(audio_file)) | |
| # elif user_input.startswith('/help'): | |
| # print_help() | |
| # else: | |
| # print("Unknown command. Type '/help' for a list of commands.") | |
| # def print_help(): | |
| # help_text = """ | |
| # 👉 To generate an image, type '/imagine <prompt>' | |
| # 👉 To chat with the model, type '/chat <prompt>' | |
| # 👉 To translate text to English, type '/translate <text>' | |
| # 👉 For Text to Speech, type '/voice <some text>' | |
| # 👉 To transcribe audio to text, type '/transcribe' and follow the prompts | |
| # 👉 For help, type '/help' | |
| # """ | |
| # print(help_text) | |
| # if __name__ == '__main__': | |
| # main() | |
| import streamlit as st | |
| from gpt_module import generate_response | |
| from dalle_module import generate_image | |
| from translation_module import translation_to_english | |
| from tts_module import text_to_speech | |
| from stt import transcribe_audio | |
| import os | |
| def main(): | |
| st.sidebar.title("Hey Rehan Assistant") | |
| user_input = st.sidebar.selectbox("Select command:", ["/imagine", "/chat", "/translate", "/voice", "/transcribe", "/help"]) | |
| if user_input == "/imagine": | |
| st.subheader('Prompt To Image generation', divider='rainbow') | |
| prompt = st.text_input("Enter prompt To Generate Image:") | |
| if st.button("Generate Image"): | |
| st.image(generate_image(prompt)) | |
| elif user_input == "/chat": | |
| st.subheader('Hey Rehan AI Assistant', divider='rainbow') | |
| prompt = st.text_input("Ask anything You want to know") | |
| if st.button("Chat"): | |
| response = generate_response(prompt) | |
| st.success(response) | |
| elif user_input == "/translate": | |
| st.subheader('Translate Into English', divider='rainbow') | |
| audio_file = st.file_uploader("Upload Audio File", type=["mp3", "wav"]) | |
| if audio_file is not None: | |
| # st.audio(uploaded_file, format='audio/wav') | |
| if st.button("Translate to English"): | |
| result = translation_to_english(audio_file) | |
| st.success(result) | |
| elif user_input == "/voice": | |
| st.subheader('Text to Speech', divider='rainbow') | |
| text = st.text_input("Enter text for Text to Speech:") | |
| if st.button("Convert to Speech"): | |
| audio_bytes = text_to_speech(text) | |
| st.audio(audio_bytes, format='audio/wav') | |
| elif user_input == "/transcribe": | |
| st.subheader('Audio to Text Transcription', divider='rainbow') | |
| uploaded_file = st.file_uploader("Upload audio file", type=["mp3", "wav"]) | |
| if uploaded_file is not None: | |
| if st.button("Transcribe Audio"): | |
| with open("temp_audio_file.mp3", "wb") as f: | |
| f.write(uploaded_file.getvalue()) | |
| transcription = transcribe_audio("temp_audio_file.mp3") | |
| st.success(transcription) | |
| os.remove("temp_audio_file.mp3") | |
| elif user_input == "/help": | |
| print_help() | |
| def print_help(): | |
| help_text = """ | |
| 👉 To generate an image, select '/imagine' and enter a prompt. | |
| 👉 To chat with the model, select '/chat' and enter a prompt. | |
| 👉 To translate text to English, select '/translate' and enter text. | |
| 👉 For Text to Speech, select '/voice' and enter some text. | |
| 👉 To transcribe audio to text, select '/transcribe' and upload the audio file. | |
| 👉 For help, select '/help'. | |
| """ | |
| st.write(help_text) | |
| if __name__ == '__main__': | |
| main() | |