Update app.py

app.py

@@ -7,30 +7,50 @@ model_name = "HuggingFaceTB/SmolLM-135M-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-def generate_feedback(text):
+# Initialize feedback history
+feedback_history = []
+
+def generate_unique_feedback(text):
     # Generate a prompt for writing feedback
     prompt = f"Provide constructive feedback on the following creative writing piece:\n\n{text}\n\nFeedback:"
     
     # Tokenize the input
     inputs = tokenizer(prompt, return_tensors="pt")
     
-    # Generate the response
-    with torch.no_grad():
-        outputs = model.generate(inputs.input_ids, do_sample=True, top_p=0.85, temperature=0.7)
-    
-    # Decode the response
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    feedback = response.split("Feedback:")[-1].strip()  # Extract feedback portion
-    return feedback
+    # Attempt to generate unique feedback
+    for _ in range(5):  # Try up to 5 times to get unique feedback
+        with torch.no_grad():
+            outputs = model.generate(inputs.input_ids, max_length=300, do_sample=True, top_p=0.85, temperature=0.7)
+        
+        # Decode the response
+        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        feedback = response.split("Feedback:")[-1].strip()
+        
+        # Check if feedback is unique
+        if feedback not in feedback_history:
+            feedback_history.append(feedback)  # Add new feedback to history
+            return feedback
+
+    # If all generated feedback is repeated, return a default message
+    return "No new feedback available at this time. Try rephrasing or adding more details to your text."
+
+# Reset history function for Gradio button
+def reset_history():
+    global feedback_history
+    feedback_history = []
+    return "Feedback history has been reset."
 
 # Set up the Gradio interface with the updated syntax
 app = gr.Interface(
-    fn=generate_feedback,
+    fn=generate_unique_feedback,
     inputs=gr.Textbox(lines=5, label="Your Writing", placeholder="Paste a short piece of creative writing here..."),
     outputs=gr.Textbox(label="Feedback"),
     title="WriteBetter",
-    description="Quick feedback on tone, grammar, and word choice."
+    description="Quick feedback on tone, grammar, and word choice.",
 )
 
+# Add a reset button to clear feedback history
+app = app.add_component(gr.Button("Reset Feedback History"), reset_history)
+
 # Launch the app
-app.launch()
+app.launch()