charlesfeng1 commited on
Commit
c7d3eea
·
verified ·
1 Parent(s): 3fafa82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -6,6 +6,7 @@ import yaml
6
  from tools.final_answer import FinalAnswerTool
7
  from tools.visit_webpage import VisitWebpageTool
8
  import os
 
9
  from urllib.parse import quote
10
 
11
  from Gradio_UI import GradioUI
@@ -76,15 +77,9 @@ def dicebear_pixel_avatar(seed: str,
76
  resp = requests.get(url, timeout=20)
77
  resp.raise_for_status()
78
 
79
- out_dir = "outputs"
80
- os.makedirs(out_dir, exist_ok=True)
81
- safe_seed = "".join(c for c in seed if c.isalnum() or c in ("-", "_"))[:40] or "seed"
82
- filename = os.path.join(out_dir, f"dicebear_pixel_{safe_seed}_{size}.{fmt}")
83
-
84
- with open(filename, "wb") as f:
85
- f.write(resp.content)
86
-
87
- return filename
88
  except Exception as e:
89
  return f"Error generating avatar: {e}"
90
 
 
6
  from tools.final_answer import FinalAnswerTool
7
  from tools.visit_webpage import VisitWebpageTool
8
  import os
9
+ import base64
10
  from urllib.parse import quote
11
 
12
  from Gradio_UI import GradioUI
 
77
  resp = requests.get(url, timeout=20)
78
  resp.raise_for_status()
79
 
80
+ # Encode image in base64
81
+ encoded = base64.b64encode(resp.content).decode("utf-8")
82
+ return f"data:image/{fmt};base64,{encoded}"
 
 
 
 
 
 
83
  except Exception as e:
84
  return f"Error generating avatar: {e}"
85