Update app.py

app.py

@@ -6,12 +6,34 @@ hti = Html2Image()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 
-def get_screenshot(html,css=None):
-
-    html = html
+html_user="""
+<div class="user_chat">
+{chat}
+</div>
+"""
+html_bot="""
+<div class="bot_chat">
+{chat}
+</div>
+"""
+html_card="""
+<div class="chat_card">
+{body}
+</div>
+"""
+
+
+def get_screenshot(chat,css=None):
+    html_body=""
+    for user,bot in history:
+        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)
     #css = css
     #hti.screenshot(html_str=html, save_as='red_page.png')
-    hti.screenshot(html_str=html, css_str=css, save_as='red_page.png')
+    hti.screenshot(html_str=html_card, css_str=css, save_as='red_page.png')
     return 'red_page.png'