Update app.py

app.py

@@ -61,32 +61,29 @@ except Exception as e:
     st.error(f"Details: {str(e)}")
 
 # Helper Functions
-def adjust_sentence_structure(output_text, tone):
+def adjust_sentence_structure(output_text):
     sentences = output_text.split(". ")
     adjusted_sentences = []
-
     for sentence in sentences:
         words = sentence.split()
-        if len(words) > 15:
+        if len(words) > 20:  # Split overly long sentences
             split_point = random.randint(len(words) // 3, 2 * len(words) // 3)
             adjusted_sentences.append(" ".join(words[:split_point]))
             adjusted_sentences.append(" ".join(words[split_point:]))
         else:
             adjusted_sentences.append(sentence)
-
     return ". ".join(adjusted_sentences)
 
 def check_grammar(output_text):
     if tool is None:
         return output_text
     matches = tool.check(output_text)
-    corrected_text = tool.correct(output_text)
-    return corrected_text
+    return tool.correct(output_text)
 
 def refine_humanization(output_text, tone, iteration):
     refinement_prompt = (
-        f"Refine the following text to sound professional and natural, avoiding repetitive patterns. "
-        f"Ensure it matches a {tone} tone. Here's the text: {output_text}"
+        f"Refine the following text to sound natural, human-like, and professional. "
+        f"Preserve the original meaning, avoid irrelevant details, and align with a {tone} tone: {output_text}"
     )
     try:
         refinement_response = client.chat.completions.create(
@@ -96,9 +93,9 @@ def refine_humanization(output_text, tone, iteration):
         )
         refined_text = refinement_response.choices[0].message.content.strip()
 
-        # Apply structure adjustments in later iterations
+        # Apply structural adjustments after the first iteration
         if iteration > 1:
-            refined_text = adjust_sentence_structure(refined_text, tone)
+            refined_text = adjust_sentence_structure(refined_text)
 
         return check_grammar(refined_text)
     except Exception:
@@ -145,20 +142,15 @@ if st.button("Generate Output"):
                 output_chunks = []
 
                 for chunk in text_chunks:
-                    if task_option == "Humanize Text":
-                        task_prompt = (
-                            f"Make the following text engaging and natural in a {tone} tone. "
-                            f"Focus on clarity and avoid robotic patterns: {chunk}"
-                        )
-                    else:
-                        task_prompt = f"Rephrase the following text while maintaining its original meaning: {chunk}"
-
+                    task_prompt = (
+                        f"{'Humanize' if task_option == 'Humanize Text' else 'Rephrase'} "
+                        f"this text while ensuring clarity, professionalism, and original meaning: {chunk}"
+                    )
                     chat_completion = client.chat.completions.create(
                         messages=[{"role": "user", "content": task_prompt}],
                         model="llama-3.3-70b-versatile",
                         stream=False,
                     )
-
                     output_text = chat_completion.choices[0].message.content.strip()
                     for i in range(humanization_depth):
                         output_text = refine_humanization(output_text, tone, i + 1)