Update app.py

app.py

@@ -5,7 +5,8 @@ import requests
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-model = genai.GenerativeModel("gemini-1.5-pro-latest")
+# model = genai.GenerativeModel("gemini-1.5-pro-latest")
+
 prompt = """
 you are a english teacher.so, understand the following Tanglish query and give a correct English sentance with proper grammar:
 Tanglish query: {tamil_text}
@@ -27,7 +28,13 @@ note: output must in one line
 
 def generate_response(input_text, prompt):
   query = prompt.format(tamil_text=input_text)
-  # model = genai.GenerativeModel("gemini-1.5-pro-latest")
+  model = genai.GenerativeModel("gemini-1.5-pro-latest")
+  response = model.generate_content(query)
+  return response.text
+
+def voice_response(input_text, prompt):
+  query = prompt.format(tamil_text=input_text)
+  model = genai.GenerativeModel("gemini-1.5-flash-latest")
   response = model.generate_content(query)
   return response.text
 
@@ -44,12 +51,12 @@ def txt2speech(text):
   
 
 st.title("🏫 English Teaching AI")
-example_text = "park ku pooga epdi english la vazhi kekuradhu?"
+example_text = "park ku pooga epdi vazhi kekuradhu?"
 user_query = st.text_area("Type Tamil or Tanglish sentance", value=example_text)
 submit = st.button("Analyze")
 if submit:
     with st.spinner("### 🤖Processing..."):
-        answer = generate_response(user_query, prompt_1)
+        answer = voice_response(user_query, prompt_1)
         txt2speech(f"In English: You can say, {answer}")
         st.audio("audio_answer.mp3")
     with st.spinner("### 🤖Analyzing your Query..."):