Update app.py

app.py

@@ -10,29 +10,34 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 ## paramas
 
 def fit_text(text:str):
-    ## prompts
-    system_prompt = 'You are a smart input form for Dall-E model'
-    user_prompt = f'''It receives the text from the user {text}, translates it into English, 
-    and reorganizes the text to make the Dall-E model understand it and use it easily to create a good image. 
-    '''
-    ## messages
+    ## Optimized system prompt
+    system_prompt = 'Generate an image based on the provided description.'
+
+    user_prompt = f'''Given the input text "{text}", please generate an image that embodies the following characteristics: 
+    - Clear and detailed representation
+    - Relevant to the provided description
+    - Incorporating key elements and context
+    - Creative and visually appealing'''
+
+    ## Messages
     messages = [ 
         {'role': 'system', 'content': system_prompt}, 
         {'role': 'user', 'content': user_prompt}
-            ]
+    ]
 
-    ## Using ChatGPT for ChatCompletion
+    ## Using ChatGPT for Chat Completion
     response = openai.ChatCompletion.create(
                 model='gpt-3.5-turbo',           
                 messages=messages,
                 temperature=0.7,  
                 max_tokens=1000,  
-                    )
+    )
 
     response = response['choices'][0]['message']['content']  
 
     return response
 
+
 def generate_image(description:str):
     response = openai.Image.create(
                 prompt=description,   ## The description for the image