Upload app.py with huggingface_hub

app.py

@@ -31,23 +31,6 @@ model = PeftModel.from_pretrained(model, LORA_MODEL, token=HF_TOKEN)
 model.eval()
 print("Model loaded!")
 
-def extract_json(text):
-    json_match = re.search(r'```(?:json)?\s*([\s\S]*?)\s*```', text)
-    if json_match:
-        try:
-            parsed = json.loads(json_match.group(1))
-            return json.dumps(parsed, indent=2, ensure_ascii=False)
-        except:
-            return json_match.group(1)
-    json_match = re.search(r'(\{[\s\S]*\})', text)
-    if json_match:
-        try:
-            parsed = json.loads(json_match.group(1))
-            return json.dumps(parsed, indent=2, ensure_ascii=False)
-        except:
-            pass
-    return ""
-
 def respond(message, history):
     messages = [{"role": "system", "content": "Du bist Zurd n8n Builder. Generiere n8n Workflow JSON in ```json Code-Bloecken."}]
     for h in history:
@@ -68,31 +51,18 @@ def respond(message, history):
         pad_token_id=tokenizer.eos_token_id
     )
     response = tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)
-    json_code = extract_json(response)
-    return response, json_code
-
-with gr.Blocks(title="Zurd n8n Builder", theme=gr.themes.Soft(primary_hue="orange")) as demo:
-    gr.Markdown("# Zurd n8n Builder\nKI-Assistent fuer n8n Workflow Automatisierung")
-
-    with gr.Row():
-        with gr.Column():
-            chatbot = gr.Chatbot(height=400, label="Chat")
-            msg = gr.Textbox(placeholder="Beschreibe deinen Workflow...", label="Nachricht")
-            with gr.Row():
-                submit = gr.Button("Senden", variant="primary")
-                clear = gr.Button("Leeren")
+    return response
 
-        with gr.Column():
-            gr.Markdown("### Workflow JSON")
-            json_output = gr.Textbox(label="JSON", lines=18, max_lines=25)
-
-    def chat(message, history):
-        response, json_code = respond(message, history)
-        history.append((message, response))
-        return "", history, json_code
-
-    submit.click(chat, [msg, chatbot], [msg, chatbot, json_output])
-    msg.submit(chat, [msg, chatbot], [msg, chatbot, json_output])
-    clear.click(lambda: ([], ""), outputs=[chatbot, json_output])
+demo = gr.ChatInterface(
+    respond,
+    title="Zurd n8n Builder",
+    description="KI-Assistent fuer n8n Workflow Automatisierung. Beschreibe deinen gewuenschten Workflow und erhalte n8n JSON.",
+    examples=[
+        "Erstelle einen Workflow der taeglich eine E-Mail mit dem Wetter sendet",
+        "Automatisierung: GitHub Issues -> Slack Benachrichtigung",
+        "API Daten abrufen und in Google Sheets speichern"
+    ],
+    theme=gr.themes.Soft(primary_hue="orange")
+)
 
 demo.launch()