Sarvam_m_indian_chat

Sleeping

App Files Files Community

prakhardoneria commited on May 22

Commit

2ea1871

verified ·

1 Parent(s): 227c037

Update app.py

Browse files

Files changed (1) hide show

app.py +33 -29

app.py CHANGED Viewed

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 MODEL_NAME = "google/flan-t5-base"
@@ -8,46 +9,49 @@ model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 def generate_code(prompt, history):
     history = history or []
-    input_text = f"Generate HTML, CSS, and JavaScript code for: {prompt}"
-    inputs = tokenizer(input_text, return_tensors="pt", truncation=True, max_length=512)
     outputs = model.generate(**inputs, max_new_tokens=512)
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    html = css = js = ""
-    if "<html>" in decoded and "</html>" in decoded:
-        html = decoded.split("<html>")[1].split("</html>")[0]
-        html = f"<html>{html}</html>"
-    if "<style>" in decoded and "</style>" in decoded:
-        css = decoded.split("<style>")[1].split("</style>")[0]
-    if "<script>" in decoded and "</script>" in decoded:
-        js = decoded.split("<script>")[1].split("</script>")[0]
-    if not (html or css or js):
-        html = decoded
-    history += [prompt, decoded]
-    return html.strip(), css.strip(), js.strip(), history
-def clear_chat():
     return "", "", "", []
 with gr.Blocks() as demo:
-    gr.Markdown(f"# Code Generator using {MODEL_NAME}")
-    with gr.Row():
-        prompt = gr.Textbox(label="Prompt", placeholder="Describe your web component...")
-        generate_btn = gr.Button("Generate")
-        new_chat_btn = gr.Button("New Chat")
     with gr.Row():
-        html_output = gr.Code(label="HTML")
-        css_output = gr.Code(label="CSS")
-        js_output = gr.Code(label="JavaScript")
-    chat_state = gr.State([])
-    generate_btn.click(fn=generate_code, inputs=[prompt, chat_state],
-                       outputs=[html_output, css_output, js_output, chat_state])
-    new_chat_btn.click(fn=clear_chat,
-                       outputs=[html_output, css_output, js_output, chat_state])
 demo.launch()

 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import json
 MODEL_NAME = "google/flan-t5-base"
 def generate_code(prompt, history):
     history = history or []
+    system_instruction = (
+        "You are a coding assistant. "
+        "Respond only in valid JSON format like this:\n"
+        "{\n"
+        "  \"filename\": \"index.html\",\n"
+        "  \"html\": \"...HTML code...\",\n"
+        "  \"css\": \"...CSS code...\",\n"
+        "  \"js\": \"...JavaScript code...\"\n"
+        "}\n"
+    )
+    full_prompt = system_instruction + "\nPrompt: " + prompt
+    inputs = tokenizer(full_prompt, return_tensors="pt", truncation=True, max_length=512)
     outputs = model.generate(**inputs, max_new_tokens=512)
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    try:
+        parsed = json.loads(decoded)
+        html = parsed.get("html", "")
+        css = parsed.get("css", "")
+        js = parsed.get("js", "")
+    except Exception:
+        html, css, js = "", "", ""
+    history.append((prompt, decoded))
+    return html, css, js, history
+def clear_history():
     return "", "", "", []
 with gr.Blocks() as demo:
+    chat_history = gr.State([])
     with gr.Row():
+        inp = gr.Textbox(label="Prompt", lines=2)
+        send_btn = gr.Button("Generate")
+        clear_btn = gr.Button("New Chat")
+    with gr.Row():
+        html_out = gr.Code(label="HTML", language="html")
+        css_out = gr.Code(label="CSS", language="css")
+        js_out = gr.Code(label="JavaScript", language="javascript")
+    send_btn.click(generate_code, [inp, chat_history], [html_out, css_out, js_out, chat_history])
+    clear_btn.click(clear_history, outputs=[html_out, css_out, js_out])
 demo.launch()