Spaces:
Sleeping
Sleeping
import assemblyai as aai | |
import google.generativeai as genai | |
import os | |
def audioTranscribe(audio_file,translation): | |
gemini_api_key = os.getenv("GEMINI_API_KEY") | |
genai.configure(api_key=gemini_api_key) | |
model = genai.GenerativeModel("gemini-1.5-pro-002") | |
aai_api_key = os.getenv("ASSEMBLYAI_API_KEY") | |
aai.settings.api_key = aai_api_key | |
transcriber = aai.Transcriber() | |
transcript = transcriber.transcribe(audio_file) | |
if translation == "French": | |
prompt = "I want you to simply translate this text into French. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Spanish": | |
prompt = "I want you to simply translate this text into Spanish. Do not elaborate on the text" | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "German": | |
prompt = "I want you to simply translate this text into German. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Italian": | |
prompt = "I want you to simply translate this text into Italian. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Japanese": | |
prompt = "I want you to simply translate this text into Japanese. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Portuguese": | |
prompt = "I want you to simply translate this text into Portuguese. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Dutch": | |
prompt = "I want you to simply translate this text into Dutch. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Korean": | |
prompt = "I want you to simply translate this text into Korean. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Hindi": | |
prompt = "I want you to simply translate this text into Hindi. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Igbo": | |
prompt = "I want you to simply translate this text into Igbo. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Yoruba": | |
prompt = "I want you to simply translate this text into Yoruba. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Polish": | |
prompt = "I want you to simply translate this text into Polish. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Swahili": | |
prompt = "I want you to simply translate this text into Swahili. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Turkish": | |
prompt = "I want you to simply translate this text into Turkish. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
elif translation == "Hausa": | |
prompt = "I want you to simply translate this text into Hausa. Do not elaborate on the text." | |
response = model.generate_content([transcript.text,prompt]) | |
return [transcript.text, response.text] | |