Update app.py

app.py

@@ -8,17 +8,17 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 html_user="""
 <div class="user_chat">
-{chat}
+$chat
 </div>
 """
 html_bot="""
 <div class="bot_chat">
-{chat}
+$chat
 </div>
 """
 html_card="""
 <div class="chat_card">
-{body}
+$body
 </div>
 """
 
@@ -29,9 +29,9 @@ def get_screenshot(chat,css=None):
     for user,bot in chat:
         print (f'user::{user}')
         print (f'bot::{bot}')
-        html_body += html_user.format(chat=user)
-        html_body += html_bot.format(chat=bot)
-    html=html_card.format(body=html_body)
+        html_body += html_user.replace("$chat",user)
+        html_body += html_bot.replace("$chat",bot)
+    html=html_card.replace("$body",html_body)
     #css = css
     #hti.screenshot(html_str=html, save_as='red_page.png')
     #hti.screenshot(html_str=html_card, css_str=css, save_as='red_page.png')