File size: 5,195 Bytes
695ae39
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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()