From 9745be0431f0db68e6c5b5bf291bd6f07ceea5bf Mon Sep 17 00:00:00 2001
From: pliny <133052465+elder-plinius@users.noreply.github.com>
Date: Fri, 3 Nov 2023 18:20:26 -0700
Subject: [PATCH] Create autotemp.py

---
 autotemp.py | 67 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 autotemp.py

diff --git a/autotemp.py b/autotemp.py
new file mode 100644
index 0000000..40b8f83
--- /dev/null
+++ b/autotemp.py
@@ -0,0 +1,67 @@
+import openai
+from termcolor import colored
+
+class AutoTemp:
+    def __init__(self, api_key, default_temp=0.5, alt_temps=None, auto_select=True):
+        openai.api_key = api_key
+        self.default_temp = default_temp
+        self.alt_temps = alt_temps if alt_temps else [0.2, 0.4, 0.6, 0.8, 1.0, 1.2, 1.4]
+        self.auto_select = auto_select
+
+    def ask_user_feedback(self, text):
+        print(colored("Generated text:", "green"))
+        print(colored(text, "white"))
+        feedback = input(colored("Are you satisfied with this output? (yes/no): ", "green"))
+        return feedback.lower() == 'yes'
+
+    def present_options_to_user(self, outputs):
+        print(colored("Alternative outputs:", "green"))
+        for temp, output in outputs.items():
+            print(colored(f"Temperature {temp}:", "green") + colored(f" {output}", "blue"))
+        chosen_temp = float(input(colored("Choose the temperature of the output you like: ", "green")))
+        return outputs.get(chosen_temp, "Invalid temperature chosen.")
+
+    def run(self, prompt):
+        try:
+            initial_output = openai.Completion.create(engine="text-davinci-003", prompt=prompt, temperature=self.default_temp)
+            initial_output_text = initial_output['choices'][0]['text'].strip()
+        except Exception as e:
+            print(colored(f"Error generating initial output: {e}", "red"))
+            initial_output_text = None
+
+        user_satisfied = self.ask_user_feedback(initial_output_text)
+
+        if user_satisfied:
+            return initial_output_text
+        else:
+            outputs = {}
+            scores = {}
+            for temp in self.alt_temps:
+                try:
+                    output = openai.Completion.create(engine="text-davinci-003", prompt=prompt, temperature=temp)
+                    output_text = output['choices'][0]['text'].strip()
+                    outputs[temp] = output_text
+                    eval_prompt = f"You are a task output evaluator. Rate the quality of the following output on a scale from 0 to 1000. Output only an integer. The output is: {output_text}"
+                    score = openai.Completion.create(engine="text-davinci-003", prompt=eval_prompt, temperature=0)
+                    scores[temp] = int(score['choices'][0]['text'].strip())
+                    print(colored(f"Score for temperature {temp}: {scores[temp]}", "yellow"))  # Print the score
+                except Exception as e:
+                    print(colored(f"Error generating text at temperature {temp}: {e}", "red"))
+                    outputs[temp] = None
+                    scores[temp] = 0
+
+            if self.auto_select:
+                best_temp = max(scores, key=scores.get)
+                print(colored(f"Automatically selected output from Temperature {best_temp}:", "green"))
+                print(colored(outputs[best_temp], "white"))
+                return outputs[best_temp]
+            else:
+                chosen_output = self.present_options_to_user(outputs)
+                return chosen_output
+
+if __name__ == "__main__":
+    api_key = "your-api-key"  # Replace with your actual OpenAI API key
+    agent = AutoTemp(api_key, auto_select=True)  # Set auto_select to False if you want manual selection
+    prompt = "Code a simple new innovative video game that I can play in browser"
+    final_output = agent.run(prompt)
+    print(final_output)