Update app.py

app.py

@@ -6,8 +6,6 @@ url = 'https://paidapi-proxy003-webraftai.run-us-west2.goorm.app/v1/chat/complet
 st.set_page_config(page_title="WebraftAI Chat", page_icon=":robot_face:")
 st.markdown("<h1 style='text-align: center;'>WebraftAI Chat</h1>", unsafe_allow_html=True)
 
-
-
 if 'generated' not in st.session_state:
     st.session_state['generated'] = []
 if 'past' not in st.session_state:
@@ -87,19 +85,20 @@ def evaluate(model, question, max_tokens=4096, top_p=0.7,temperature=0.1):
             sentence = response['choices'][0]['message']['content']
         else:
             sentie = response
-            sentence = f"API Returned: ```{sentie}``` \n Is the API not working? check https://stats.uptimerobot.com/KkZE2T84mj \n If you are still encountering errors then please open a ticket from our discord server at https://discord.gg/webraftai . "
+            sentence = f"API Returned: \n ```{sentie}``` \n \n Is the API not working? check https://stats.uptimerobot.com/KkZE2T84mj \n If you are still encountering errors then please open a ticket from our discord server at https://discord.gg/webraftai . "
 
 
     except requests.exceptions.JSONDecodeError as e:
         sentie= response.content
-        sentence = f"API Returned: ```{sentie}``` \n Is the API not working? check https://stats.uptimerobot.com/KkZE2T84mj \n If you are still encountering errors then please open a ticket from our discord server at https://discord.gg/webraftai . "
+        sentence = f"API Returned:\n ```{sentie}``` \n \n Is the API not working? check https://stats.uptimerobot.com/KkZE2T84mj \n If you are still encountering errors then please open a ticket from our discord server at https://discord.gg/webraftai . "
     return sentence
 
 
 def generate_response(prompt):
     st.session_state['messages'].append({"role": "user", "content": prompt})
+    prompt = prompt.lower()
     if prompt.startswith("draw"):
-        data = {"prompt": prompt,
+        data = {"prompt": prompt.replace("draw",""),
                 "model": "realistic-vision-v1-4"
                 }
 
@@ -111,12 +110,12 @@ def generate_response(prompt):
 
             response = response.json()
             imageurl = response['data'][0]['url']
-            ie = f"Here is the generated image: {imageurl} \n Model used: `realistic-vision-v1-4`"
+            ie = f"Here is the generated image:\n {imageurl} \n \n Model used: `realistic-vision-v1-4`"
             st.session_state['messages'].append({"role": "assistant", "content": ie})
 
         except requests.exceptions.JSONDecodeError as e:
             sentie = response.content
-            ie= f"Image generation API returned: ```{sentie}``` \n Is the API not working? check https://stats.uptimerobot.com/KkZE2T84mj \n If you are still encountering errors then please open a ticket from our discord server at https://discord.gg/webraftai . "
+            ie= f"Image generation API returned:\n ```{sentie}``` \n \n Is the API not working? check https://stats.uptimerobot.com/KkZE2T84mj \n If you are still encountering errors then please open a ticket from our discord server at https://discord.gg/webraftai . "
             st.session_state['messages'].append({"role": "assistant", "content": ie})
         return ie,total_tokens,prompt_tokens,completion_tokens