Update app.py

app.py

@@ -55,28 +55,37 @@ def query_chatgpt(inputs, history, message):
     return history, history, ""
 
 
-def initialize_prompt(prompt_type, history):
+def initialize_prompt(history):
     history = history or []
 
-    if prompt_type:
-        prompt_file = "./prompts/" + str(prompt_type) + ".txt"
-
-        with open(prompt_file, "r") as f:
-            prompt = f.read()
-        output = ask(prompt)
-        history.append(("<ORIGINAL PROMPT>", output))
+    output = ask('Hello ChatGPT!')
+    history.append(("<ORIGINAL PROMPT>", output))
 
     return history, history
 
 
+def display_prompt(show, prompt_type):
+    if not prompt_type:
+        show = False
+        return "Error - prompt not selected"
+
+    else:
+        if show:
+            prompt_file = "./prompts/" + str(prompt_type) + ".txt"
+
+            with open(prompt_file, "r") as f:
+                prompt = f.read()
+
+            return prompt
+        else:
+            return ""
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("""<h3><center>ChatGPT + Robotics</center></h3>""")
+    gr.Markdown("""<h3><center>ChatGPT Hacked</center></h3>""")
     gr.Markdown(
-        """This is a companion app to the work [ChatGPT for Robotics: Design Principles and Model Abilities](https://aka.ms/ChatGPT-Robotics).<br>
-        This space allows you to work with ChatGPT to get it to generate code for robotics tasks, such as get a robot arm to manipulate objects, or have a drone fly around in a 3D world.<br>  
-        See [README](https://huggingface.co/spaces/microsoft/ChatGPT-Robotics/blob/main/README.md) for detailed instructions."""
+        """This is a hacked version of the original ChatGPT demo. It is a work in progress.
+        """
     )
 
     if not access_token:
@@ -91,6 +100,8 @@ with gr.Blocks() as demo:
     l = os.listdir("./prompts")
     li = [x.split(".")[0] for x in l]
 
+    gr.Markdown("""<h4>Initial Prompt for ChatGPT</h4>""")
+
     initialize = gr.Button(value="Initialize")
 
     gr.Markdown("""<h4>Conversation</h4>""")
@@ -103,7 +114,7 @@ with gr.Blocks() as demo:
 
     state = gr.State()
 
-    initialize.click(fn=initialize_prompt, inputs=[prompt_type, state], outputs=[chatgpt_robot, state])
+    initialize.click(fn=initialize_prompt, inputs=[state], outputs=[chatgpt_robot, state])
 
     message.submit(query_chatgpt, inputs=[message, state], outputs=[chatgpt_robot, state, message])