Update app.py

app.py

@@ -33,36 +33,38 @@ new_emotions_prompt = [
 ]
 
 import os
-from transformers import Prompt, SaveModel
-import gr
-from datasets import load_dataset, Dataset
+
+from google.colab import files
+from google.colab import drive
 
 def emo(name, temper, save_response):
-    prompters = Prompt(model_name="gpt-3.5-turbo", temper=temper)
+    prompters = Prompter("gpt-3.5-turbo", temper)
     response_text = prompters.prompt_model_print(new_emotions_prompt)
     if save_response:
-        save_emotion_output(response_text)
+        save_response_to_file(response_text)
+    return response_text
 
 def save_emotion_output(response_text):
-    model_name = "gpt-3.5-turbo"
-    model = SaveModel.from_pretrained(model_name)
-    model.save_pretrained("emotion_output.txt")
-    print("Saved emotion output to emotion_output.txt")
+    with open("/content/drive/MyDrive/emotion_output.html", "a") as file:
+        file.write(response_text)
+        file.write("\n")
+    print("Saved emotion output to emotion_output.html")
 
 with gr.Blocks() as demo:
     with gr.Row():
-        slider = gr.Slider(0, 1, label="Temperature",
+        slider = gr.Slider(0, 1, label='Temperature',
                            info="Choose a float value between 0 and 1 to set the 'creativity' "
-                           "of the machine. Keep in mind that a value of 1 (highest) may create "
-                           "inconsistent but also interesting results.")
+                                "of the machine. Keep in mind that a value of 1 (highest) may create inconsistent "
+                                "but also interesting results.")
         emo_out = gr.Textbox(value='text', label="Output", lines=3)
         #save_btn = gr.Button(label="Save Emotion")
-    #with gr.Row():
-    #    btn = gr.Button(value="Generate Emotion")
-    #    btn.click(emo, inputs=slider, outputs = emo_out)
-    #save_btn = gr.Button(value="Save Emotion")
-    #    btn.click(save_emotion_output, inputs=emo_out)
+    with gr.Row():
+      btn = gr.Button(value="Generate Emotion")
+      btn.click(emo, inputs=slider, outputs = emo_out)
+      save_btn = gr.Button(value="Save Emotion")
+      save_btn.click(save_emotion_output, inputs=emo_out)
+  
+    if __name__ == "__main__":
+        demo.launch()
 
-if __name__ == "__main__":
-    demo.launch()