Update app.py

app.py

@@ -82,17 +82,21 @@ def dubpro_english_transliteration(text):
     return response_content
 
 
-def generate_rephrases_gemini(text, language):
+def generate_rephrases_gemini(text, language, problem):
     API_URL = os.environ.get("GEMINI_REPHRASER_API")
     BEARER_TOKEN = get_google_token()
     headers = {
         "Authorization": f"Bearer {BEARER_TOKEN}",
         "Content-Type": "application/json",
     }
+    if problem == "Gap":
+        speak = "more"
+    else:
+        speak = "less"
     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."
+        prompt = f"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 {speak} 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."
+        prompt = f"You are a hindi language expert please rephrase the below line without summary so that it takes little {speak} time to speak in hinglish manner."
         
     payload = {
         "contents": [
@@ -128,7 +132,10 @@ def generate_rephrases_gemini(text, language):
     output_text = response["candidates"][0]["content"]["parts"][0]["text"]
 
     texts = list(map(lambda x: x.replace("-", "").strip(), output_text.split("\n")))
-    return "\n".join(texts)
+
+    wc = f"Original Word Count: {len(text.split())}\nRephrased Word Count: {len(texts.split())}"
+    
+    return "\n".join(texts), wc
 
 
 with gr.Blocks() as demo:
@@ -143,9 +150,12 @@ with gr.Blocks() as demo:
     with gr.Row():
         rephrase_text = gr.Textbox(label="Input text", info="Please enter text.")
         language = gr.Dropdown(["English", "Hindi"], value="Hindi")
+        solving_for = gr.Dropdown(["Gap", "Overflow"], value="Overflow", label="Solving for:")
+    with gr.Row():
+        word_count = gr.Textbox(label="Word count")
         rephrased_text = gr.Textbox(label="Output text")
     rephrase = gr.Button("Submit")
-    rephrase.click(generate_rephrases_gemini, [rephrase_text, language], rephrased_text)
+    rephrase.click(generate_rephrases_gemini, [rephrase_text, language, solving_for], [rephrased_text, word_count])
         
 
 demo.launch(auth=(os.environ.get("USERNAME"), os.environ.get("PASSWORD")))