Update app.py

app.py

@@ -45,7 +45,7 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
     prompt = alpaca_instruct_prompt.format(system_str, input_text,"")
     
     output = gen_pipeline(
-        input_text,
+        prompt,
         max_new_tokens=max_new_tokens,
         temperature=temperature,
         return_full_text = False,
@@ -62,9 +62,20 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
         generated_text = "Spiacente, non sono in grado di rispondere."
         generated_text_color = 'red'
     
-    return f"<span>{input_text}</span><b style='color: {generated_text_color};'>{generated_text}</b>"
+    return f"<span>{input_text}</span><b style='color: {generated_text_color};'>{generated_text}</b>", {"input_prompt":prompt, "generated_text_raw":output[0]['generated_text'], "generated_text_displayed":generated_text}
+
+def positive_feedback(last_generated_text):
+    print("positive")
+    print(last_generated_text)
+
+def negative_feedback(last_generated_text):
+    print("negative")
+    print(last_generated_text)
+    
 
 # Create the Gradio interface
+last_generated_text = gr.State({"input_prompt":"", "generated_text_raw":"", "generated_text_displayed":""})
+
 input_text = gr.Textbox(lines=2, placeholder="Enter your request here...", label="Input Text")
 system_prompt = gr.Textbox(lines=2, placeholder="Enter custom system prompt...", label="Custom System Prompt")
 
@@ -83,7 +94,12 @@ with gr.Blocks(css="#outbox { border-radius: 8px !important; border: 1px solid #
     output = gr.HTML(label="Generated Text",elem_id="outbox")
 
     btn = gr.Button("Generate")
-    btn.click(generate_text, [input_text, max_new_tokens, temperature, system_prompt], output)
+    btn.click(generate_text, [input_text, max_new_tokens, temperature, system_prompt], outputs=[output, last_generated_text])
+
+    btn_p = gr.Button("👍")
+    btn_p.click(positive_feedback, inputs=[last_generated_text], outputs=None)
+    btn_n = gr.Button("👎")
+    btn_n.click(negative_feedback, inputs=[last_generated_text], outputs=None)
 
 # Launch the interface
 if __name__ == "__main__":