Update app.py

app.py

@@ -3,26 +3,22 @@ from transformers import pipeline
 import random
 import logging
 
-# Set up logging
+# Enable logging
 logging.basicConfig(level=logging.INFO)
 
-# Load paraphrasing model
+# Load Hugging Face models
 paraphraser = pipeline("text2text-generation", model="t5-base", max_length=512)
-
-# Load grammar correction model
 grammar_corrector = pipeline("text2text-generation", model="vennify/t5-base-grammar-correction", max_length=512)
 
-# Function to rephrase content
 def rephrase_content(text: str) -> str:
     try:
-        prompt = f"paraphrase: {text} </s>"
-        result = paraphraser(prompt, do_sample=True, top_k=50, top_p=0.95)[0]
-        return result["generated_text"]
+        prompt = f"paraphrase: {text}"
+        result = paraphraser(prompt, do_sample=True, top_k=50, top_p=0.95)[0]['generated_text']
+        return result.strip()
     except Exception as e:
-        logging.error(f"Paraphrasing failed: {e}")
+        logging.error(f"Error in rephrasing: {e}")
         return text
 
-# Function to inject critical thinking phrase
 def inject_critical_thinking(text: str) -> str:
     critical_phrases = [
         "However, this perspective may be limited due to...",
@@ -33,33 +29,30 @@ def inject_critical_thinking(text: str) -> str:
     ]
     return f"{text} {random.choice(critical_phrases)}"
 
-# Function to correct grammar
 def correct_grammar(text: str) -> str:
     try:
         prompt = f"grammar: {text}"
-        result = grammar_corrector(prompt)[0]
-        return result["generated_text"]
+        result = grammar_corrector(prompt)[0]['generated_text']
+        return result.strip()
     except Exception as e:
-        logging.error(f"Grammar correction failed: {e}")
+        logging.error(f"Error in grammar correction: {e}")
         return text
 
-# Main processing function
 def process_text(text: str) -> str:
-    logging.info("Processing text...")
+    logging.info("Starting academic text enhancement...")
     rephrased = rephrase_content(text)
     injected = inject_critical_thinking(rephrased)
     corrected = correct_grammar(injected)
-    return corrected.strip()
+    logging.info("Final output generated.")
+    return corrected
 
-# Gradio interface
+# Gradio UI
 iface = gr.Interface(
     fn=process_text,
-    inputs=gr.Textbox(lines=15, label="Input your academic content"),
-    outputs=gr.Textbox(lines=15, label="Processed academic content"),
-    title="Academic Enhancer API",
-    description="Rephrases, injects critical thinking, and fixes grammar for academic writing."
+    inputs=gr.Textbox(lines=15, label="Paste your academic content"),
+    outputs=gr.Textbox(lines=15, label="Enhanced Content"),
+    title="Academic Writing Enhancer",
+    description="Rephrase, enhance and correct academic content for clarity and originality."
 )
 
-iface.launch(server_name="0.0.0.0",
-    server_port=7860,
-    show_api=True) 
+iface.launch()