Update app.py

app.py

@@ -82,12 +82,64 @@ def dubpro_english_transliteration(text):
     return response_content
 
 
+def generate_rephrases_gemini(text):
+    API_URL = os.environ.get("GEMINI_FINETUNED_HINDI_ENG_API")
+    BEARER_TOKEN = get_google_token()
+    headers = {
+        "Authorization": f"Bearer {BEARER_TOKEN}",
+        "Content-Type": "application/json",
+    }
+    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": f"input: {text}"
+                    },
+                    {
+                        "text": f"output: "
+                    }
+                ],
+                "role": "user",
+            }
+        ],
+        "generationConfig": {
+            "maxOutputTokens": 8192,
+            "temperature": 0.85,
+            "candidateCount": 1,
+        },
+        "safetySettings": [
+            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
+            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
+        ],
+    }
+    result = requests.post(url=API_URL, headers=headers, json=payload)
+    response = result.json()
+    output_text = response["candidates"][0]["content"]["parts"][0]["text"]
+
+    texts = list(map(lambda x: x.replace("-", "").strip(), output_text.split("\n")))
+    return texts
+
+
 with gr.Blocks() as demo:
     gr.Markdown("English Transliteration Tool")
     with gr.Row():
         input_text = gr.Textbox(label="Input text", info="Please enter English text.")
         output_text = gr.Textbox(label="Output text")
-    submit = gr.Button("Submit")
-    submit.click(dubpro_english_transliteration, input_text, output_text)
+    transliterate = gr.Button("Submit")
+    transliterate.click(dubpro_english_transliteration, input_text, output_text)
+
+    gr.Markdown("Rephraser Tool")
+    with gr.Row():
+        rephrase_text = gr.Textbox(label="Input text", info="Please enter text.")
+        rephrased_text = gr.Textbox(label="Output text")
+    rephrase = gr.Button("Submit")
+    rephrase.click(generate_rephrases_gemini, rephrase_text, rephrased_text)
+        
 
 demo.launch(auth=(os.environ.get("USERNAME"), os.environ.get("PASSWORD")))