Upload app.py

app.py

@@ -2,14 +2,19 @@
 import os
 import csv
 import openai
+from openai import OpenAI
 import gradio as gr
 import huggingface_hub
 from huggingface_hub import Repository
 from datetime import datetime
 
-# set up for the Hugging Face dataset, based on the directory you set up
-DATASET_REPO_URL = "https://huggingface.co/datasets/jadend/mr_bot_requests"
-DATA_FILENAME = "premise_input.csv"
+client = OpenAI(
+    api_key=os.environ.get("API_TOKEN"),
+)
+
+# recording requests to Hugging Face dataset
+DATASET_REPO_URL = "https://huggingface.co/datasets/petcoblue/simulation_data"
+DATA_FILENAME = "user_agents.csv"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -19,12 +24,12 @@ repo = Repository(
 )
 
 # function to store the message to the Hugging Face site, if you would like to add more columns or information to track add it here
-def store_message(premise: str, temperature: str, response: str):
-    if premise:
+def store_message(prompt: str, temperature: str, response: str):
+    if prompt:
         with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["premise", "temperature", "response", "time"])
+            writer = csv.DictWriter(csvfile, fieldnames=["prompt", "temperature", "response", "time"])
             writer.writerow(
-                {"premise": premise, 
+                {"prompt": prompt, 
                  "temperature": temperature,
                  "response": response,
                  "time": str(datetime.now())}
@@ -34,40 +39,65 @@ def store_message(premise: str, temperature: str, response: str):
 
     return 
 
-# defines who can enter the application with the secrets that are set up
+# defines who can enter the application with the secrets that are set up if used
 user_db = { 
            os.environ["username"]: os.environ["password"],
           }
 
 # the OpenAI function to use the given prompt to the model defined.
-def mr_bot(premise, tokens, temperature):
-    openai.api_key = os.environ["API_TOKEN"]
-    response = openai.Completion.create(
-        model="text-davinci-003",
-        prompt=premise,
+def mr_bot(prompt, tokens, temperature):
+    response = client.chat.completions.create(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": prompt}],
         max_tokens=tokens,
         temperature=temperature,
         )
-    store_message(premise=premise, temperature=temperature, response=response['choices'][0]["text"])
+    store_message(prompt=prompt, temperature=temperature, response=response.choices[0].message.content)
     
-    return response['choices'][0]["text"]
+    return response.choices[0].message.content
+
 
 # Gradio App interface and login set up
-description = "Ask Mr. Bot a question or give him a command. Entered Prompts are logged and stored securely."
+description = "Use the examples below to see how different prompts generate different results. Use the examples to see how different prompting " \
+              "techniques can enhance your prompts. They are split into three sections, (creative writing, problem solving in business, and " \
+              "philosophical debate) with three different levels of prompt engineering, (basic, intermediate, and advanced)."
 
 demo = gr.Interface(fn=mr_bot, 
-                    inputs=["text", gr.Slider(1, 800, default=800), gr.Slider(0.1, 1, default=0.75)], 
+                    inputs=[gr.Textbox(placeholder="Select an example", interactive=False), gr.Slider(1, 800, default=800), gr.Slider(0.1, 1, default=0.75)], 
                     outputs="text",
-                    title="Mr. Bot",
+                    title="Intro to AI, Prompt Engineering",
                     description=description,
                     cache_examples=False,
-                    examples = [["Write me a rap song about how the price of rent is too high.", 800, .75],
-                               ["Make a weekly meal plan for me comprised of an entirely vegan diet.", 800, .50]]
+                    examples = [
+                                            # Creative Writing
+                        # Basic Prompt:
+                        ["Write a story about a city.", 800, .90],
+                        # Intermediate Prompt:
+                        ["Write a story set in a futuristic city where technology controls everything.", 800, .90],
+                        # Advanced Prompt:
+                        ["Write a gripping tale set in a futuristic city named Neo-Philly, where an underground movement " \
+                         "is fighting against the oppressive control of AI over human life, focusing on a protagonist who was once a loyal AI prompt engineer.", 800, .90],
+
+                                            # Problem Solving in Business
+                        # Basic Prompt:
+                        ["A company needs to improve its profits.", 800, .90],
+                        # Intermediate Prompt:
+                        ["A retail company is seeing a decline in profits due to online competition. What can they do?", 800, .90],
+                        # Advanced Prompt:
+                        ["Devise a comprehensive strategy for a brick-and-mortar retail company that has been losing market share to e-commerce giants. " \
+                         "Focus on innovative in-store experiences, digital integration, and customer loyalty programs to regain competitiveness.", 800, .90],
+
+                                            # Philosophical Debate
+                        # Basic Prompt:
+                        ["Discuss the concept of happiness.", 800, .90],
+                        # Intermediate Prompt:
+                        ["Debate whether true happiness can be achieved through material wealth or if it's found in intangible experiences.", 800, .90],
+                        # Advanced Prompt:
+                        ["Critically analyze the philosophical arguments for and against the notion that true happiness is derived more from personal " \
+                         "fulfillment and self-actualization than from material possessions, considering perspectives from both Eastern and " \
+                         "Western philosophies.", 800, .90],
+                    ]
                     )
 
 if __name__ == "__main__":
-    demo.launch(enable_queue=False,
-        auth=lambda u, p: user_db.get(u) == p,
-        auth_message="Welcome!"
-               )
-                
+    demo.launch(enable_queue=False)