added prompt random

main.py

@@ -9,14 +9,15 @@ client = OpenAI()
 import streamlit as st
 from PIL import Image
 from diffusers import AutoPipelineForText2Image
-
+import random
 @st.cache_data(ttl=600)
 def get_prompt_to_guess():
+    random_prompt = ["tree", "cat", "dog", "consultant", "artificial intelligence", "beauty", "immeuble", "plage", "cyborg", "futuristic"]
     response = client.chat.completions.create(
     model="gpt-3.5-turbo",
     messages=[
         {"role": "system", "content": "You are a helpful assistant to generate one simple prompt in order to generate an image. Your given prompt won't go over 10 words. You only return the prompt. You will also answer in french."},
-        {"role": "user", "content": "Donne moi un prompt pour generer une image"},
+        {"role": "user", "content": f"Donne moi un prompt pour generer une image de {random.choice(random_prompt)}"},
     ]
     )
     return response.choices[0].message.content