Update app.py

app.py

@@ -1,57 +1,89 @@
+!pip install transformers gradio sentencepiece
+
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
-import torch
 
 # Load model and tokenizer
 model_name = "ramsrigouthamg/t5_paraphraser"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
+# Define style prompts
+def generate_prompt(text, style):
+    style_prompts = {
+        "Formal": "Please rewrite the following text in a formal and professional tone:\n\n",
+        "Friendly": "Please rewrite the following text in a casual and friendly tone:\n\n",
+        "Poetic": "Rewrite the text in a poetic and metaphorical way, like a short verse:\n\n",
+        "Gen Z": "Rewrite this text using Gen Z slang, internet expressions, abbreviations, and emojis:\n\n"
+    }
+
+    # Default if style not found
+    base_prompt = style_prompts.get(style, "Rewrite the text:\n\n")
+    return base_prompt + text
+
+# Function to rewrite text in selected style
 def rewrite_text(text, style):
     if not text.strip():
         return "Please enter some text."
 
-    style_prompts = {
-        "Formal": "Rewrite this formally: ",
-        "Friendly": "Rewrite this casually and friendly: ",
-        "Poetic": "Rewrite this like a poem: ",
-        "Gen Z": "Rewrite this like a Gen Z person using slang and emojis: ",
-        "Professional": "Rewrite this in a professional tone: ",
-        "Witty": "Rewrite this in a witty and clever way: "
-    }
+    # Build the prompt based on the selected style
+    if style == "Gen Z":
+        prompt = f"Rewrite the following text in a funny Gen Z tone with slang, emojis, and internet expressions:\n\n{text}"
+    elif style == "Poetic":
+        prompt = f"Rewrite the following text in a poetic and artistic style:\n\n{text}"
+    elif style == "Formal":
+        prompt = f"Rewrite the following text in a formal, professional tone:\n\n{text}"
+    elif style == "Friendly":
+        prompt = f"Rewrite the following text in a friendly and conversational style:\n\n{text}"
+    else:
+        prompt = f"Rewrite the following text:\n\n{text}"
 
-    prompt = style_prompts.get(style, "Paraphrase: ") + text
+    # Call the model (paraphraser)
+    response = paraphraser(prompt, max_length=100)[0]['generated_text']
 
-    input_ids = tokenizer.encode(prompt, return_tensors="pt", max_length=256, truncation=True)
-    output_ids = model.generate(input_ids, max_length=100, num_beams=5, early_stopping=True)
-    output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+    # Optional: clean output (remove repeated prompt from response if needed)
+    return response.replace(prompt, "").strip()
 
-    return output
 
 # Gradio UI
+import gradio as gr
+
+# Define the interface
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     gr.Markdown(
         """
         <h1 style="text-align: center;">📝 Rewrite My Text</h1>
-        <p style="text-align: center;">Transform your text into <b>Formal, Friendly, Poetic, Gen Z, Witty</b> styles using AI! 🚀</p>
-        """
+        <p style="text-align: center;">Transform your text into <b>fun, formal, poetic, or Gen Z</b> styles using AI! 🚀</p>
+        """,
+        elem_id="header",
     )
 
     with gr.Row():
-        with gr.Column():
-            input_text = gr.Textbox(label="Input Text", placeholder="Enter your sentence...", lines=4)
+        with gr.Column(scale=1):
+            input_text = gr.Textbox(
+                label="Input Text",
+                placeholder="Enter your sentence here...",
+                lines=4
+            )
             style = gr.Dropdown(
                 label="Choose Style",
-                choices=["Gen Z", "Formal", "Poetic", "Friendly", "Professional", "Witty"],
+                choices=["Gen Z", "Formal", "Poetic", "Friendly"],
                 value="Gen Z"
             )
             submit_button = gr.Button("✨ Submit", variant="primary")
             clear_button = gr.Button("🧹 Clear")
 
-        with gr.Column():
-            output_text = gr.Textbox(label="Rewritten Text", lines=4)
+        with gr.Column(scale=1):
+            output_text = gr.Textbox(
+                label="Rewritten Text",
+                placeholder="Your rewritten sentence will appear here...",
+                lines=4
+            )
 
+    # Button functionality
     submit_button.click(fn=rewrite_text, inputs=[input_text, style], outputs=output_text)
     clear_button.click(fn=lambda: ("", ""), inputs=[], outputs=[input_text, output_text])
 
-demo.launch()
+# Launch the app
+demo.launch(share=True)
+