Update app.py

app.py

@@ -10,9 +10,11 @@ css="""
     background: #367bb9;
     border-radius: 10px 0px 10px 10px;
     padding: 10px;
-    margin-bottom: 10px;
     max-width: 100%;
     text-wrap: pretty;
+    margin-left:10px;
+    margin-bottom: 10px;
+
 }
 .bot_chat{
     background: #1c3c9b;
@@ -20,6 +22,7 @@ css="""
     padding: 10px;
     max-width: 100%;
     text-wrap: pretty;
+    margin-right:10px;
     margin-bottom: 10px;
     
 }
@@ -66,6 +69,7 @@ def get_screenshot(chat):
     #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')
+    hti.size = (500, 1000)
     hti.screenshot(html_str=html, css_str=css, save_as='red_page.png')
     return 'red_page.png',html