Update app.py

app.py

@@ -57,22 +57,27 @@ $body
 </pre>
 """
 
-
+def make_filename(inp):
+    if len(inp)<=20:
+        first_name=inp
+    else:
+        first_name=inp[:20]
+    print(first_name)
+    last_name=random.randint(1,1111111111)
+    filename=first_name+last_name
+    return filename
 def get_screenshot(chat,css_in,w,h):
-    print(chat)
+    
     html_body=""
     for user,bot in chat:
-        print (f'user::{user}')
-        print (f'bot::{bot}')
         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')
     hti.size = (w, h)
-    hti.screenshot(html_str=html, css_str=css_in, save_as='red_page.png')
-    return 'red_page.png',html
+    filename=f'{make_filename(chat[0][0])}.png'
+    print(filename)
+    hti.screenshot(html_str=html, css_str=css_in, save_as=filename)
+    return filename,html
 
 
 def format_prompt(message, history):
@@ -133,7 +138,7 @@ with gr.Blocks(css=css) as app:
         with gr.Column(scale=1):
             with gr.Group():
                 with gr.Row():
-                    im_height=gr.Number(label="Height",value=500)
+                    im_height=gr.Number(label="Height",value=1000)
                     im_width=gr.Number(label="Width",value=500)
                 wait_time=gr.Number(label="Wait Time",value=3000)
                 theme=gr.Radio(label="Theme", choices=["light","dark"],value="light")