fix error: Invalid type for messages[0]

app.py

@@ -78,8 +78,8 @@ with gr.Blocks(fill_height=True) as demo:
 
     # Input fields for initialization
     task_input = gr.Textbox(label="Task", placeholder="Describe your task...")
-    app_name_input = gr.Textbox(label="App Name", placeholder="Enter the app name...")
-    sitemap_input = gr.Textbox(label="Sitemap", placeholder="Enter the Hugging Face link to sitemap...")
+    app_name_input = gr.Textbox(label="App Name", placeholder="Enter the app name... (eg. DoorDash)")
+    sitemap_input = gr.Textbox(label="Sitemap", placeholder="Enter the Hugging Face link to sitemap... (eg.jjz5463/DoorDash_synthetic_sitemap)")
     initialize_button = gr.Button("Initialize Simulator")
 
     # Status block to display initialization progress with elapsed time

chatbot_simulator.py

@@ -77,13 +77,26 @@ Rules:
     def _get_openai_response(self, prompt):
         """Fetch response from OpenAI API."""
         self._trim_conversation()
-        response = self.client.chat.completions.create(
-            model="gpt-4",
-            messages=prompt,
-            max_tokens=self.buffer_tokens,  # Adjusted max_tokens if needed
-            temperature=0.7,
-        )
-        return response.choices[0].message.content
+        while True:
+            try:
+                response = self.client.chat.completions.create(
+                    model="gpt-4",
+                    messages=prompt,
+                    max_tokens=self.buffer_tokens,  # Adjusted max_tokens if needed
+                    temperature=0.7,
+                )
+                return response.choices[0].message.content
+            except RateLimitError as e:
+                # Parse the suggested retry time from the error message, default to 5s if not available
+                wait_time = 5
+                try:
+                    # Attempt to get the time from the error message
+                    wait_time = float(e.response['error']['message'].split("in ")[1].split("s")[0])
+                except (KeyError, IndexError, ValueError):
+                    print("Could not parse wait time from error message. Defaulting to 5 seconds.")
+
+                print(f"Rate limit reached. Retrying in {wait_time} seconds...")
+                time.sleep(wait_time)
 
     def _calculate_token_count(self, conversation):
         """Accurately calculate the token count in the conversation using a tokenizer."""