File size: 5,195 Bytes
572c05d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2b51f30
 
 
 
 
 
 
 
 
 
7f42a4d
 
b2c602a
 
2b51f30
b2c602a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6c17c87
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b2c602a
572c05d
 
 
 
 
 
 
 
 
 
 
 
b2c602a
572c05d
 
 
 
 
2b51f30
6c17c87
572c05d
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
import os
import random

import gradio as gr

#links
Article = ""
Article = Article + " Gallery UI documentation: https://gradio.app/docs/"
Article = Article + " Unsplash free image source requires 5MP or larger images:  https://unsplash.com/"
Article = Article + " Github should also be a worthy alternative with free repos.  "
Article = Article + " Raw git content can be accessed by URL like so:  https://raw.github.com/AaronCWacker/Yggdrasil/images/"
Article = Article + " Originals go here: https://github.com/AaronCWacker/Yggdrasil/tree/main/images"
# Aaron_Wacker_health_and_medical_icon_set_on_white_background_bba24b60-9fcf-411b-9c00-dd1ba1e3553c.png


import os
import csv
import gradio as gr
from gradio import inputs, outputs
import huggingface_hub
from huggingface_hub import Repository
from datetime import datetime

HF_TOKEN = os.environ.get("HF_TOKEN")
print("is none?", HF_TOKEN is None)
DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/BookComposer"
DATA_FILENAME = "BookComposer.csv"
DATA_FILE=os.path.join("data", DATA_FILENAME)
HF_TOKEN=os.environ.get("HF_TOKEN")

def gan():
    images = [
        (random.choice(
            [
                "Comectress.png",
                "Comic Plate 1 - Amnesia.png",
                "Comic Plate 1 - Depths.png",
                "Comic Plate 1 - Helion.png",
                "Comic Plate 1 - Kitsune.png",
                "Comic Plate 1 - Sinbad.png",
                "Comic Plate 1 - Vampiress.png",
                "Comic Plate 2 - Amnesia.png",
                "Comic Plate 2 - Depths.png",
                "Comic Plate 2 - Helion.png",
                "Comic Plate 2 - Kitsune.png",
                "Comic Plate 2 - Sinbad.png",
                "Comic Plate 2 - Vampiress.png",
                "Comic Plate 3 - Amnesia.png",
                "Comic Plate 3 - Depths.png",
                "Comic Plate 3 - Helion.png",
                "Comic Plate 3 - Kitsune.png",
                "Comic Plate 3 - Vampiress.png",
                "Comic Plate 4 - Amnesia.png",
                "Comic Plate 4 - Depths.png",
                "Comic Plate 4 - Helion.png",
                "Comic Plate 4 - Kitsune.png",
                "Comic Plate 4 - Sinbad.png",
                "Comic Plate 4 - Vampiress.png",
                "Comic Plate 5 - Vampiress.png",
                "Gold Suit.png",
                "Heavens.png",
                "Red Dot Sight.png",
                "Starfire.png",
                "Vamp 1.png",
                "Vamp 2.png",
                "Vamp 3.png",
                "Vamp 4.png",
                "Whirlwind.png",
                "Zyphoria.png",
            ]
        ), f"label {i}" if i != 0 else "label" * 50)
        for i in range(10)
    ]
    return images

def gan2():
    images = [
        (random.choice(
            [
                "Comectress.png",
                "Comic Plate 1 - Amnesia.png",
                "Comic Plate 1 - Depths.png",
                "Comic Plate 1 - Helion.png",
                "Comic Plate 1 - Kitsune.png",
                "Comic Plate 1 - Sinbad.png",
                "Comic Plate 1 - Vampiress.png",
                "Comic Plate 2 - Amnesia.png",
                "Comic Plate 2 - Depths.png",
                "Comic Plate 2 - Helion.png",
                "Comic Plate 2 - Kitsune.png",
                "Comic Plate 2 - Sinbad.png",
                "Comic Plate 2 - Vampiress.png",
                "Comic Plate 3 - Amnesia.png",
                "Comic Plate 3 - Depths.png",
                "Comic Plate 3 - Helion.png",
                "Comic Plate 3 - Kitsune.png",
                "Comic Plate 3 - Vampiress.png",
                "Comic Plate 4 - Amnesia.png",
                "Comic Plate 4 - Depths.png",
                "Comic Plate 4 - Helion.png",
                "Comic Plate 4 - Kitsune.png",
                "Comic Plate 4 - Sinbad.png",
                "Comic Plate 4 - Vampiress.png",
                "Comic Plate 5 - Vampiress.png",
                "Gold Suit.png",
                "Heavens.png",
                "Red Dot Sight.png",
                "Starfire.png",
                "Vamp 1.png",
                "Vamp 2.png",
                "Vamp 3.png",
                "Vamp 4.png",
                "Whirlwind.png",
                "Zyphoria.png",
            ]
        ), f"label {i}" if i != 0 else "label" * 50)
        for i in range(10)
    ]
    return images



with gr.Blocks() as demo:
    with gr.Column(variant="panel"):
        with gr.Row(variant="compact"):
            text = gr.Textbox(
                label="Health and Medical Icon Sets",
                show_label=False,
                max_lines=1,
                placeholder="Enter your prompt",
            ).style(
                container=False,
            )
            btn = gr.Button("Generate image").style(full_width=False)
            btn2 = gr.Button("Generate story").style(full_width=False)

        gallery = gr.Gallery(
            label="Generated images", show_label=False, elem_id="gallery"
        ).style(grid=[2], height="auto")

    btn.click(gan, None, gallery)
    btn2.click(gan2, None, gallery)

if __name__ == "__main__":
    demo.launch()