Spaces:

Rahul-8799
/

Multi_agent_test

Runtime error

App Files Files Community

Rahul-8799 commited on May 10

Commit

30d86ab

verified ·

1 Parent(s): f8536c8

Upload langgraph_pipeline.py

Browse files

Files changed (1) hide show

utils/langgraph_pipeline.py +97 -57

utils/langgraph_pipeline.py CHANGED Viewed

@@ -1,4 +1,4 @@
-import uuid, zipfile, re
 from pathlib import Path
 from typing import TypedDict, List, Dict, Any, Tuple
@@ -157,70 +157,110 @@ def parse_spec(spec: str) -> Dict[str, List[str]]:
 def run_pipeline_and_save(prompt: str) -> Tuple[List[Dict[str, Any]], str]:
     # a) invoke agents
     initial_state = {"messages": [HumanMessage(content=prompt)], "chat_log": [], "iteration": 0, "feedback": ""}
-    final_state   = compiled_graph.invoke(initial_state)
-    chat_log  = final_state["chat_log"]
-    qa_output = final_state["qa_output"]
-    # b) parse spec
-    spec = parse_spec(qa_output)
-    features     = spec.get("Key features", [])
-    testimonials = spec.get("User stories", [])
-    # c) build HTML
-    title = prompt.title()
-    domain = prompt.replace(" ", "").lower() + ".com"
-    cards_html = "\n".join(f"<div class='card'><h3>{f}</h3></div>" for f in features)
-    test_html  = "\n".join(f"<blockquote>{t}</blockquote>" for t in testimonials)
-    html_code = f"""<!DOCTYPE html>
-<html lang="en">
-<head>
-  <meta charset="UTF-8">
-  <meta name="viewport" content="width=device-width,initial-scale=1">
-  <title>{title}</title>
-  <link rel="stylesheet" href="styles.css">
-</head>
-<body>
-  <header><h1>{title}</h1></header>
-  <section id="features">
-    <h2>Features</h2>
-    <div class="cards">
-      {cards_html}
-    </div>
-  </section>
-  <section id="testimonials">
-    <h2>Testimonials</h2>
-    {test_html or '<p>No testimonials provided.</p>'}
-  </section>
-  <section id="contact">
-    <h2>Contact Us</h2>
-    <p>Email: info@{domain}</p>
-  </section>
-</body>
-</html>"""
-    # d) basic CSS
-    css_code = """
-body { font-family: Arial, sans-serif; margin: 1em; line-height: 1.5; }
-header { text-align: center; margin-bottom: 2em; }
-.cards { display: grid; grid-template-columns: repeat(auto-fit,minmax(150px,1fr)); gap: 1em; }
-.card { background: #f9f9f9; padding: 1em; border-radius: 8px; box-shadow: 0 2px 5px rgba(0,0,0,0.1); text-align: center; }
-blockquote { font-style: italic; margin: 1em; padding: 0.5em; background: #eef; border-left: 4px solid #99f; }
-"""
-    # e) write & zip
-    site_id  = uuid.uuid4().hex
-    out_dir  = Path("output")
     site_dir = out_dir / f"site_{site_id}"
     site_dir.mkdir(parents=True, exist_ok=True)
-    (site_dir / "index.html").write_text(html_code, encoding="utf-8")
-    (site_dir / "styles.css").write_text(css_code,  encoding="utf-8")
     zip_path = out_dir / f"site_{site_id}.zip"
     with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
         for f in site_dir.iterdir():
             zf.write(f, arcname=f.name)
     return chat_log, str(zip_path)

+import uuid, zipfile, re, json
 from pathlib import Path
 from typing import TypedDict, List, Dict, Any, Tuple
 def run_pipeline_and_save(prompt: str) -> Tuple[List[Dict[str, Any]], str]:
     # a) invoke agents
     initial_state = {"messages": [HumanMessage(content=prompt)], "chat_log": [], "iteration": 0, "feedback": ""}
+    final_state = compiled_graph.invoke(initial_state)
+    chat_log = final_state["chat_log"]
+    dev_output = final_state["dev_output"]
+    # b) parse the developer output to extract code sections
+    sections = parse_code_sections(dev_output)
+    # c) write & zip
+    site_id = uuid.uuid4().hex
+    out_dir = Path("output")
     site_dir = out_dir / f"site_{site_id}"
     site_dir.mkdir(parents=True, exist_ok=True)
+    # Write HTML file
+    (site_dir / "index.html").write_text(sections.get("HTML Structure", ""), encoding="utf-8")
+    # Write CSS file
+    (site_dir / "styles.css").write_text(sections.get("CSS Styles", ""), encoding="utf-8")
+    # Write JavaScript file
+    (site_dir / "script.js").write_text(sections.get("JavaScript", ""), encoding="utf-8")
+    # Write Tailwind config
+    (site_dir / "tailwind.config.js").write_text(sections.get("Tailwind Config", ""), encoding="utf-8")
+    # Create package.json for dependencies
+    package_json = {
+        "name": f"site_{site_id}",
+        "version": "1.0.0",
+        "description": "Generated responsive website",
+        "scripts": {
+            "build": "tailwindcss -i ./styles.css -o ./dist/output.css",
+            "watch": "tailwindcss -i ./styles.css -o ./dist/output.css --watch"
+        },
+        "dependencies": {
+            "tailwindcss": "^3.4.1",
+            "alpinejs": "^3.13.3"
+        }
+    }
+    (site_dir / "package.json").write_text(
+        json.dumps(package_json, indent=2),
+        encoding="utf-8"
+    )
+    # Create README
+    readme_content = f"""# Generated Website
+This is a responsive website generated by the Multi-Agent UI Generator.
+## Setup
+1. Install dependencies:
+   ```bash
+   npm install
+   ```
+2. Build the CSS:
+   ```bash
+   npm run build
+   ```
+3. For development with live reload:
+   ```bash
+   npm run watch
+   ```
+## Features
+- Responsive design using Tailwind CSS
+- Interactive elements with JavaScript
+- Modern animations and transitions
+- Mobile-first approach
+"""
+    (site_dir / "README.md").write_text(readme_content, encoding="utf-8")
+    # Create zip file
     zip_path = out_dir / f"site_{site_id}.zip"
     with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
         for f in site_dir.iterdir():
             zf.write(f, arcname=f.name)
     return chat_log, str(zip_path)
+def parse_code_sections(output: str) -> Dict[str, str]:
+    """Parse code sections from the developer output"""
+    sections = {}
+    current_section = None
+    current_code = []
+    for line in output.split("\n"):
+        if line.startswith("## "):
+            if current_section:
+                sections[current_section] = "\n".join(current_code)
+            current_section = line[3:].strip()
+            current_code = []
+        elif line.startswith("```"):
+            continue
+        elif current_section:
+            current_code.append(line)
+    if current_section:
+        sections[current_section] = "\n".join(current_code)
+    return sections