Spaces:

spandana30
/

Multi_agent

Build error

App Files Files Community

spandana30 commited on Apr 7

Commit

cd1de35

verified ·

1 Parent(s): 3748474

Upload 3 files

Browse files

Files changed (3) hide show

README (1).md +33 -0
app.py +138 -0
requirements.txt +3 -0

README (1).md ADDED Viewed

	@@ -0,0 +1,33 @@

+---
+title: Multi-Agent UI Generator
+emoji: 🤝
+colorFrom: indigo
+colorTo: teal
+sdk: streamlit
+sdk_version: "1.32.0"
+app_file: app.py
+pinned: false
+---
+# Multi-Agent UI Generator 🚀
+This is a Streamlit-based collaborative system where agents (Designer, Software Engineer, QA) work together to generate production-ready UI code.
+## Features
+- Natural language prompt → Design specs
+- Generate complete HTML/CSS UI from specs
+- QA feedback loop until UI is approved
+- Live preview and downloads for HTML + logs
+## Instructions
+1. Create a Hugging Face Space (Streamlit SDK)
+2. Upload `app.py`, `requirements.txt`, and this `README.md`
+3. Set your Hugging Face token in Space Secrets as `HF_TOKEN`
+## Powered by
+- LangGraph 🧠
+- Hugging Face Inference API 💬
+- Streamlit 🚀

app.py ADDED Viewed

	@@ -0,0 +1,138 @@

+import streamlit as st
+import os
+import time
+import random
+import base64
+from typing import Dict, List, TypedDict
+from io import StringIO
+from langgraph.graph import StateGraph, END
+from huggingface_hub import InferenceClient
+client = InferenceClient(
+    model="mistralai/Mistral-7B-Instruct-v0.2",
+    token=st.secrets["HF_TOKEN"]
+)
+class AgentState(TypedDict):
+    messages: List[Dict[str, str]]
+    design_specs: str
+    html: str
+    css: str
+    feedback: str
+    iteration: int
+    done: bool
+    timings: Dict[str, float]
+DESIGNER_PROMPT = """You're a UI designer. Create design specs for:
+{user_request}
+Include:
+1. Color palette (primary, secondary, accent)
+2. Font choices
+3. Layout structure
+4. Component styles
+Don't write code - just design guidance."""
+ENGINEER_PROMPT = """Create a complete HTML page with embedded CSS for:
+{design_specs}
+Requirements:
+1. Full HTML document with <!DOCTYPE>
+2. CSS inside <style> tags in head
+3. Mobile-responsive
+4. Semantic HTML
+5. Ready-to-use (will work when saved as .html)
+Output JUST the complete HTML file content:"""
+QA_PROMPT = """Review this website:
+{html}
+Check for:
+1. Visual quality
+2. Responsiveness
+3. Functionality
+Reply "APPROVED" if perfect, or suggest improvements."""
+def time_agent(agent_func, state: AgentState, label: str):
+    start = time.time()
+    result = agent_func(state)
+    duration = time.time() - start
+    result["timings"] = state["timings"]
+    result["timings"][label] = duration
+    return result
+def designer_agent(state: AgentState):
+    specs = call_model(DESIGNER_PROMPT.format(user_request=state["messages"][-1]["content"]))
+    return {"design_specs": specs, "messages": state["messages"] + [{"role": "designer", "content": specs}]}
+def engineer_agent(state: AgentState):
+    html = call_model(ENGINEER_PROMPT.format(design_specs=state["design_specs"]))
+    if not html.strip().startswith("<!DOCTYPE"):
+        html = f"""<!DOCTYPE html>
+<html><head><meta charset='UTF-8'><meta name='viewport' content='width=device-width, initial-scale=1.0'>
+<title>Generated UI</title></head><body>{html}</body></html>"""
+    return {"html": html, "messages": state["messages"] + [{"role": "software_engineer", "content": html}]}
+def qa_agent(state: AgentState, max_iter: int):
+    feedback = call_model(QA_PROMPT.format(html=state["html"]))
+    done = "APPROVED" in feedback or state["iteration"] >= max_iter
+    return {"feedback": feedback, "done": done, "iteration": state["iteration"] + 1,
+            "messages": state["messages"] + [{"role": "qa", "content": feedback}]}
+def call_model(prompt: str, max_retries=3) -> str:
+    for attempt in range(max_retries):
+        try:
+            return client.text_generation(prompt, max_new_tokens=3000, temperature=0.3, return_full_text=False)
+        except Exception:
+            time.sleep(2)
+    return "<html><body><h1>Error generating UI</h1></body></html>"
+def generate_ui(user_request: str, max_iter: int):
+    state = {"messages": [{"role": "user", "content": user_request}], "design_specs": "", "html": "",
+             "css": "", "feedback": "", "iteration": 0, "done": False, "timings": {}}
+    workflow = StateGraph(AgentState)
+    workflow.add_node("designer", lambda s: time_agent(designer_agent, s, "designer"))
+    workflow.add_node("software_engineer", lambda s: time_agent(engineer_agent, s, "software_engineer"))
+    workflow.add_node("qa", lambda s: time_agent(lambda x: qa_agent(x, max_iter), s, "qa"))
+    workflow.add_edge("designer", "software_engineer")
+    workflow.add_edge("software_engineer", "qa")
+    workflow.add_conditional_edges("qa", lambda s: END if s["done"] else "software_engineer")
+    workflow.set_entry_point("designer")
+    app = workflow.compile()
+    total_start = time.time()
+    final_state = app.invoke(state)
+    return final_state["html"], final_state, time.time() - total_start
+def main():
+    st.set_page_config(page_title="Multi-Agent Collaboration", layout="wide")
+    st.title("🤝 Multi-Agent Collaboration")
+    with st.sidebar:
+        st.header("Settings")
+        max_iter = st.slider("Max QA Iterations", 1, 5, 2)
+    prompt = st.text_area("Describe the UI you want:", "A clean dashboard with 3 summary cards and a sidebar", height=150)
+    if st.button("Generate UI"):
+        with st.spinner("Working..."):
+            html, final_state, total_time = generate_ui(prompt, max_iter)
+            st.success("UI Generated Successfully!")
+            st.components.v1.html(html, height=600, scrolling=True)
+            st.subheader("📥 Download HTML")
+            b64 = base64.b64encode(html.encode()).decode()
+            st.markdown(f'<a href="data:file/html;base64,{b64}" download="ui.html">Download HTML</a>', unsafe_allow_html=True)
+            st.subheader("🧠 Agent Log")
+            history_text = "".join([f"---\n{m['role'].title()}:\n{m['content']}\n\n" for m in final_state["messages"]])
+            st.text_area("Conversation", history_text, height=300)
+            b64_hist = base64.b64encode(history_text.encode()).decode()
+            st.markdown(f'<a href="data:file/txt;base64,{b64_hist}" download="agent_log.txt">Download Log</a>', unsafe_allow_html=True)
+            st.subheader("📊 Performance")
+            st.write(f"Total Time: {total_time:.2f} seconds")
+            st.write(f"Iterations: {final_state['iteration']}")
+            for stage in ["designer", "software_engineer", "qa"]:
+                st.write(f"{stage.replace('_', ' ').title()}: {final_state['timings'].get(stage, 0):.2f}s")
+if __name__ == "__main__":
+    main()

requirements.txt ADDED Viewed

	@@ -0,0 +1,3 @@

+streamlit
+huggingface_hub
+langgraph