Update app.py

app.py

@@ -1,44 +1,48 @@
-import gradio as gr
-from transformers import pipeline
-
-# Load the paraphrase model
-model_name = "AventIQ-AI/t5-paraphrase-generation"
-paraphrase_pipeline = pipeline("text2text-generation", model=model_name)
-
-def generate_paraphrase(text, max_length, temperature):
-    """Generate a paraphrased version of the input text."""
-    if not text.strip():
-        return "⚠️ Please enter some text to paraphrase."
-    
-    result = paraphrase_pipeline(
-        text,
-        max_length=max_length,
-        temperature=temperature,  # Adds randomness to prevent repetition
-        top_k=50,  # Consider top-k tokens for variation
-        do_sample=True  # Enable sampling
-    )
-    return result[0]["generated_text"]
-
-# Define Gradio Interface
-description = """
-## ✨ AI Paraphrasing Tool
-Enter a sentence and let AI generate a paraphrased version!
-- Adjust **max length** for longer outputs.
-- Tune **temperature** for more creative results.
-"""
-
-demo = gr.Interface(
-    fn=generate_paraphrase,
-    inputs=[
-        gr.Textbox(label="Enter text", placeholder="Type a sentence to paraphrase..."),
-        gr.Slider(20, 100, value=50, step=5, label="Max Output Length"),
-        gr.Slider(0.5, 1.5, value=1.0, step=0.1, label="Creativity (Temperature)"),
-    ],
-    outputs=gr.Textbox(label="Paraphrased Text"),
-    title="📝 AI Paraphraser",
-    description=description,
-    theme="huggingface",
-    live=True,
-)
-
-demo.launch()
+import gradio as gr
+from transformers import pipeline
+
+# Load the paraphrase model
+model_name = "AventIQ-AI/t5-paraphrase-generation"
+paraphrase_pipeline = pipeline("text2text-generation", model=model_name)
+
+def generate_paraphrase(text, temperature):
+    """Generate a paraphrased version of the input text."""
+    if not text.strip():
+        return "⚠️ Please enter some text to paraphrase."
+    
+    # Limit input to 700 words
+    words = text.split()
+    if len(words) > 700:
+        return "⚠️ Please enter a maximum of 700 words."
+
+    result = paraphrase_pipeline(
+        text,
+        temperature=temperature,  # Adds randomness to prevent repetition
+        top_k=50,  # Consider top-k tokens for variation
+        do_sample=True  # Enable sampling
+    )
+    
+    return result[0]["generated_text"]
+
+# Define Gradio Interface
+description = """
+## ✨ AI Paraphrasing Tool
+Enter text and let AI generate a paraphrased version!
+- **Creativity (Temperature)** controls how varied the output is.
+- **Input is limited to 700 words.**
+"""
+
+demo = gr.Interface(
+    fn=generate_paraphrase,
+    inputs=[
+        gr.Textbox(label="Enter text", placeholder="Type your text to paraphrase...", lines=5),
+        gr.Slider(0.5, 1.5, value=1.0, step=0.1, label="Creativity (Temperature)"),
+    ],
+    outputs=gr.Textbox(label="Paraphrased Text"),
+    title="📝 AI Paraphraser",
+    description=description,
+    theme="huggingface",
+    live=True,
+)
+
+demo.launch()