Spaces:
Running
Running
File size: 3,474 Bytes
8fd768e a985eac 8fd768e a985eac 8fd768e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
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]
|