broadfield-dev commited on
Commit
2ebe379
·
verified ·
1 Parent(s): c63fa25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -12
app.py CHANGED
@@ -10,9 +10,8 @@ import bs4
10
  import datetime
11
  import urllib.request
12
  import uuid
13
- import os
14
- from huggingface_hub import hf_hub_url
15
-
16
 
17
 
18
 
@@ -137,20 +136,22 @@ def make_animation(url1="",url2="",url3="",url4="",url5="",url6="",url7="",url8=
137
  frames.append(f'tmp{i}.png')
138
  fps=60
139
  uid=uuid.uuid4()
140
- clip = ic.ImageSequenceClip(frames, fps = fps)
141
- clip.write_gif(f"{uid}.gif",fps=fps)
142
- files_out.append(f'{uid}.gif')
143
 
144
- #print(os.path.join(os.getcwd(), f"{uid}.gif"))
145
- #out_gif_url = os.path.join(os.getcwd(), f"{uid}.gif")
146
- url = hf_hub_url(repo_id="broadfield-dev/geoelectric", filename=f"{uid}.gif")
147
- print(url)
148
- #img_src=f"https://huggingface.co/broadfield-dev/geoelectric/blob/main/{uid}.gif"
149
 
 
 
 
 
 
150
 
 
 
 
151
 
 
152
  html_out=f"<style>{css}</style><div>"
153
- html_out+=f'<div class="img_class"><img src="{url}"></div>'
154
  html_out+="</div>"
155
  except Exception as e:
156
  print("ERROR HAPPENED")
 
10
  import datetime
11
  import urllib.request
12
  import uuid
13
+ import io
14
+ import base64
 
15
 
16
 
17
 
 
136
  frames.append(f'tmp{i}.png')
137
  fps=60
138
  uid=uuid.uuid4()
139
+ clip = ic.ImageSequenceClip(frames, fps = fps)
 
 
140
 
 
 
 
 
 
141
 
142
+ # Generate GIF in memory
143
+ gif_buffer = io.BytesIO()
144
+ clip.write_gif(gif_buffer, fps=fps)
145
+ gif_bytes = gif_buffer.getvalue()
146
+ gif_buffer.close()
147
 
148
+ # Encode GIF bytes as base64 for HTML
149
+ gif_base64 = base64.b64encode(gif_bytes).decode('utf-8')
150
+ gif_data_url = f"data:image/gif;base64,{gif_base64}"
151
 
152
+
153
  html_out=f"<style>{css}</style><div>"
154
+ html_out+=f'<div class="img_class"><img src="{gif_data_url}"></div>'
155
  html_out+="</div>"
156
  except Exception as e:
157
  print("ERROR HAPPENED")