Update app.py

app.py

@@ -82,19 +82,24 @@ def dubpro_english_transliteration(text):
     return response_content
 
 
-def generate_rephrases_gemini(text):
-    API_URL = os.environ.get("GEMINI_FINETUNED_HINDI_ENG_API")
+def generate_rephrases_gemini(text, language):
+    API_URL = os.environ.get("GEMINI_REPHRASER_API")
     BEARER_TOKEN = get_google_token()
     headers = {
         "Authorization": f"Bearer {BEARER_TOKEN}",
         "Content-Type": "application/json",
     }
+    if language == "English":
+        prompt = "You are an English and Hindi language expert, please rephrase a sentence that has been translated from Hindi to English so that it takes little less time to speak."
+    elif language == "Hindi":
+        prompt = "You are a hindi language expert please rephrase the below line without summary so that it takes little less time to speak in hinglish manner."
+        
     payload = {
         "contents": [
             {
                 "parts": [
                     {
-                        "text": "You are a language expert. Return rephrases of the provided text without any loss in information keeping the same tone and style of text, so that it takes a little less time when generating audio. Return only the outputs in a new line."
+                        "text": prompt
                     },
                     {
                         "text": f"input: {text}"
@@ -123,7 +128,7 @@ def generate_rephrases_gemini(text):
     output_text = response["candidates"][0]["content"]["parts"][0]["text"]
 
     texts = list(map(lambda x: x.replace("-", "").strip(), output_text.split("\n")))
-    return texts
+    return "\n".join(texts)
 
 
 with gr.Blocks() as demo:
@@ -137,9 +142,10 @@ with gr.Blocks() as demo:
     gr.Markdown("Rephraser Tool")
     with gr.Row():
         rephrase_text = gr.Textbox(label="Input text", info="Please enter text.")
+        language = gr.Dropdown(["English", "Hindi"], value="Hindi")
         rephrased_text = gr.Textbox(label="Output text")
     rephrase = gr.Button("Submit")
-    rephrase.click(generate_rephrases_gemini, rephrase_text, rephrased_text)
+    rephrase.click(generate_rephrases_gemini, [rephrase_text, laguage], rephrased_text)
         
 
 demo.launch(auth=(os.environ.get("USERNAME"), os.environ.get("PASSWORD")))