Spaces:
Running
Running
File size: 4,206 Bytes
87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 87727f0 b0359f9 |
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 |
import gradio as gr
from llm_models import get_text_image_pairs
import time
from tqdm import tqdm
title_markdown = """
<div style="display: flex; justify-content: center; align-items: center; text-align: center; direction: rtl;">
<img src="https://s11.ax1x.com/2023/12/28/piqvDMV.png" alt="MoE-LLaVA🚀" style="max-width: 120px; height: auto; margin-right: 20px;">
<div style="display: flex; flex-direction: column; justify-content: center; align-items: center;">
<h1 style="margin: 0; font-size: 4em;">الراوي</h1>
<br>
<h2 style="margin: 0; font-size: 1.5em;">صانع القصص بالذكاء الاصطناعي التوليدي</h2>
</div>
</div>
"""
def get_text_images_values(k, input_prompt, api_key_openai, api_key_stability_ai):
pages = int(k)
segments_list, images_names = get_text_image_pairs(
pages, input_prompt, api_key_openai, api_key_stability_ai
)
return segments_list, images_names
css = """
.gradio-container {direction: rtl}
.gradio-container-4-18-0 .prose h1 {direction: rtl};
}
"""
with gr.Blocks(css=css) as demo:
gr.Markdown(title_markdown)
with gr.Row():
api_key_openai = gr.Textbox(
label="Open AI API Key",
placeholder="أدخل مفتاح API الخاص بك هنا",
type="password",
)
api_key_stability_ai = gr.Textbox(
label="Stability AI API Key",
placeholder="أدخل مفتاح API الخاص بك هنا",
type="password",
)
prompt = gr.Textbox(
label="معلومات بسيطة عن القصة",
info="أدخل بعض المعلومات عن القصة، مثلاً: خالد صبي في الرابعة من عمره، ويحب أن يصبح طياراً في المستقبل",
placeholder="خالد صبي في الرابعة من عمره، ويحب أن يصبح طياراً في المستقبل",
text_align="right",
rtl=True,
elem_classes="rtl-textbox",
elem_id="rtl-textbox",
)
with gr.Row():
max_textboxes = 10 # Define the max number of textboxed, so we will add the max number of textboxes and images to the layout
def variable_outputs(k, segments_list):
k = int(k)
return [
gr.Textbox(
label=f"الصفحة رقم {i+1}",
value=item,
text_align="right",
visible=True,
)
for i, item in enumerate(segments_list)
] + [gr.Textbox(visible=False, text_align="right", rtl=True)] * (
max_textboxes - k
)
def variable_outputs_image(k, images_names):
k = int(k)
return [
gr.Image(value=item, scale=1, visible=True) for item in images_names
] + [gr.Image(scale=1, visible=False)] * (max_textboxes - k)
with gr.Column():
s = gr.Slider(
1,
max_textboxes,
value=1,
step=1,
info="أقصى عدد صفحات يمكن توليده هو 10 صفحات",
label="كم عدد صفحات القصة التي تريدها؟",
)
textboxes = []
imageboxes = []
for i in tqdm(range(max_textboxes)):
with gr.Row():
i_t = gr.Image(visible=False)
t = gr.Textbox(visible=False)
imageboxes.append(i_t)
textboxes.append(t)
segment_list = gr.JSON(value=[], visible=False)
images_list = gr.JSON(value=[], visible=False)
submit = gr.Button(value="أنشئ القصة الآن")
submit.click(
fn=get_text_images_values,
inputs=[s, prompt, api_key_openai, api_key_stability_ai],
outputs=[segment_list, images_list],
).then(
fn=variable_outputs,
inputs=[s, segment_list],
outputs=textboxes,
).then(
fn=variable_outputs_image,
inputs=[s, images_list],
outputs=imageboxes,
)
demo.launch()
|