log

app.py

@@ -79,18 +79,33 @@ def handler(text, ts1, ts2, gs1):
 
 text = st.text_area('asdf')
 
+ts1, ts2, gs1 = 20, 20, 0
+
 if st.button('rweerew'):
     low_res = st.empty()
     high_res = st.empty()
 
-    for s, xs in handler(text, 20, 20, 0):
+    count_low_res, count_high_res = 0, 0
+
+    for s, xs in handler(text, ts1, ts2, gs1):
         print((type(s), s.shape))
         print((type(xs), s.shape))
         s = Image.fromarray(s[0])
         xs = Image.fromarray(xs[0])
-        target = low_res if s.size[0] < 256 else high_res
+        is_high_res = s.size[0] == 256
+        if is_high_res:
+            target = high_res
+            count = count_high_res
+            total = ts2
+        else:
+            target = low_res
+            count = count_low_res
+            total = ts1
+        count += 1
         with target.container():
             st.image([s, xs])
+            st.write(f'{count} / {total}')
+
 
 # x = st.slider('Select a value')
 # st.write(x, 'squared is', x * x)