Qudrat0708 commited on
Commit
b555b36
·
verified ·
1 Parent(s): f5b52e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -13
app.py CHANGED
@@ -1,28 +1,25 @@
1
- import os
2
  import whisper
3
  from gtts import gTTS
4
  import gradio as gr
5
- from groq import Groq
6
  from googletrans import Translator # Translator for language conversion
7
 
8
- GROQ_API_KEY = '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'
 
 
9
 
10
  # Load Whisper model for transcription (use a multilingual model to support Urdu)
11
  model = whisper.load_model("large") # Use "large" or "multilingual" for better Urdu support
12
 
13
- # Set up Groq API client
14
- client = Groq(api_key=GROQ_API_KEY)
15
-
16
  # Set up Translator
17
  translator = Translator()
18
 
19
- # Function to get the LLM response from Groq
20
  def get_llm_response(user_input):
21
- chat_completion = client.chat.completions.create(
22
- messages=[{"role": "user", "content": user_input}],
23
- model="llama3-8b-8192", # Replace with your desired model
24
- )
25
- return chat_completion.choices[0].message.content
26
 
27
  # Function to convert text to speech using gTTS
28
  def text_to_speech(text, output_audio="output_audio.mp3"):
@@ -36,7 +33,7 @@ def chatbot(audio):
36
  result = model.transcribe(audio, language="ur") # Specify Urdu language
37
  user_text = result["text"]
38
 
39
- # Step 2: Get LLM response from Groq (in English)
40
  response_text = get_llm_response(user_text)
41
 
42
  # Step 3: Translate the English response to Urdu
 
1
+ import os
2
  import whisper
3
  from gtts import gTTS
4
  import gradio as gr
5
+ from googleapiclient.discovery import build
6
  from googletrans import Translator # Translator for language conversion
7
 
8
+ # Gemini API key
9
+ API_KEY = "AIzaSyAh5Q4iJYJOH6hBZnIg1Gikb_jb8qDl-x0"
10
+ service = build("gemini", "v1", developerKey=API_KEY) # Adjust service name and version if necessary
11
 
12
  # Load Whisper model for transcription (use a multilingual model to support Urdu)
13
  model = whisper.load_model("large") # Use "large" or "multilingual" for better Urdu support
14
 
 
 
 
15
  # Set up Translator
16
  translator = Translator()
17
 
18
+ # Function to get the LLM response from Gemini
19
  def get_llm_response(user_input):
20
+ # Call the Gemini API
21
+ request = service.text().generate(body={"prompt": user_input}).execute()
22
+ return request.get("generatedText", "No response generated") # Default to "No response" if empty
 
 
23
 
24
  # Function to convert text to speech using gTTS
25
  def text_to_speech(text, output_audio="output_audio.mp3"):
 
33
  result = model.transcribe(audio, language="ur") # Specify Urdu language
34
  user_text = result["text"]
35
 
36
+ # Step 2: Get LLM response from Gemini (in English)
37
  response_text = get_llm_response(user_text)
38
 
39
  # Step 3: Translate the English response to Urdu