Spaces:
Running
Running
File size: 3,549 Bytes
ee8b38f a8b7750 a786c9d a8b7750 dfd696d 3dd38da |
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 |
import gradio as gr
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
# Cargar modelo m谩s peque帽o para generar c贸digo
model_name = "Qwen/Qwen2.5-Coder-1.5B-Instruct"
model = AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype=torch.float16 # Usar float16 para ahorrar memoria
)
tokenizer = AutoTokenizer.from_pretrained(model_name)
def generate_code(prompt):
"""Genera c贸digo basado en el prompt del usuario."""
messages = [
{"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
{"role": "user", "content": prompt},
{"role": "assistant", "content": ""}
]
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
model_inputs = tokenizer([text], return_tensors="pt")
generated_ids = model.generate(
**model_inputs,
max_new_tokens=512,
do_sample=True,
temperature=0.7
)
response = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
return response
def preview_app(html_code, css_code, js_code):
"""Devuelve una vista previa interactiva de la aplicaci贸n."""
html_content = f"""
<html>
<head>
<style>
{css_code}
</style>
</head>
<body>
{html_code}
<script>
{js_code}
</script>
</body>
</html>
"""
return html_content
def run_chatbot(user_input):
"""Procesa la entrada del usuario y genera c贸digo + previsualizaci贸n."""
code_output = generate_code(user_input)
# Extraer HTML, CSS y JS del c贸digo generado
html_code = ""
css_code = ""
js_code = ""
if "<style>" in code_output:
css_start = code_output.find("<style>") + len("<style>")
css_end = code_output.find("</style>")
css_code = code_output[css_start:css_end].strip()
if "<script>" in code_output:
js_start = code_output.find("<script>") + len("<script>")
js_end = code_output.find("</script>")
js_code = code_output[js_start:js_end].strip()
if "<body>" in code_output:
html_start = code_output.find("<body>") + len("<body>")
html_end = code_output.find("</body>")
html_code = code_output[html_start:html_end].strip()
# Previsualizar la aplicaci贸n
preview = preview_app(html_code, css_code, js_code)
return (
f"### HTML:\n\n```html\n{html_code}\n```",
f"### CSS:\n\n```css\n{css_code}\n```",
f"### JavaScript:\n\n```javascript\n{js_code}\n```",
preview
)
# Crear la interfaz con Gradio
with gr.Blocks() as demo:
gr.Markdown("# Chatbot Creador de Aplicaciones")
with gr.Row():
with gr.Column():
user_input = gr.Textbox(label="Descripci贸n de la aplicaci贸n (Ejemplo: 'Haz un bot贸n rojo')", lines=3)
generate_button = gr.Button("Generar C贸digo")
with gr.Column():
html_output = gr.Code(label="C贸digo HTML", language="html")
css_output = gr.Code(label="C贸digo CSS", language="css")
js_output = gr.Code(label="C贸digo JavaScript", language="javascript")
preview_output = gr.HTML(label="Previsualizaci贸n")
generate_button.click(
run_chatbot,
inputs=[user_input],
outputs=[html_output, css_output, js_output, preview_output]
)
# Lanzar la aplicaci贸n
if __name__ == "__main__":
demo.launch() |