Spaces:

lydiasolomon
/

DevAssist

Sleeping

App Files Files Community

alaselababatunde commited on Sep 19

Commit

caa2aff

1 Parent(s): e9ae209

Updated

Browse files

Files changed (1) hide show

main.py +23 -31

main.py CHANGED Viewed

@@ -24,11 +24,16 @@ if not SPITCH_API_KEY:
 os.environ["SPITCH_API_KEY"] = SPITCH_API_KEY
 spitch_client = Spitch()
-# HuggingFace LLM
 llm = HuggingFaceEndpoint(
     repo_id=HF_MODEL,
     temperature=0.2,
-    model_kwargs={"max_length": 2048}
 )
 # FastAPI app
@@ -76,34 +81,21 @@ Documentation:
 """
 sme_template = """
-You are an SME site builder AI.
-Your job is to turn ANY user prompt (simple or complex) into a working, modern web project.
-You must analyze the request carefully and generate clean, professional, and responsive code.
-Rules:
-1. Always return ONLY valid JSON (no explanations, no Markdown).
-2. Always include "index.html", "style.css", and "script.js" in the "files".
-3. Support multiple pages if the user specifies them.
-4. Style must be modern, vibrant, and responsive:
-   - Use semantic HTML5 (<header>, <main>, <section>, <footer>).
-   - Use CSS with modern fonts, colors, spacing, hover effects, flex/grid.
-   - Buttons must be styled and interactive.
-   - Add responsiveness for mobile.
-5. Translate vague descriptions like “make it futuristic and sharp” into concrete CSS design choices.
-6. If the request is very detailed, honor as much as possible while still keeping valid code.
-User Prompt:
-{user_prompt}
-Return ONLY JSON in this format:
-{
-  "files": {
-    "index.html": "<!DOCTYPE html> ... </html>",
-    "style.css": "/* modern responsive CSS here */",
-    "script.js": "// optional interactivity here"
-  },
-  "README.md": "# Instructions on how to run the project"
-}
 """
 # ----------------- CHAINS -----------------
@@ -142,7 +134,7 @@ def chat(req: ChatRequest, authorization: str | None = Header(None)):
         return {"reply": answer.strip() if isinstance(answer, str) else str(answer)}
     except HfHubHTTPError as e:
         if "exceeded" in str(e).lower() or "quota" in str(e).lower():
-            return {"reply": "⚠️ You have reached your daily limit for responses. Please come back in 24 hours."}
         raise e
 @app.post("/stt")

 os.environ["SPITCH_API_KEY"] = SPITCH_API_KEY
 spitch_client = Spitch()
+# HuggingFace LLM (better tuned for code generation)
 llm = HuggingFaceEndpoint(
     repo_id=HF_MODEL,
     temperature=0.2,
+    model_kwargs={
+        "max_new_tokens": 1500,   # more space for multi-file generation
+        "do_sample": True,
+        "top_p": 0.9,
+        "repetition_penalty": 1.1
+    }
 )
 # FastAPI app
 """
 sme_template = """
+You are a senior full-stack engineer specializing in modern front-end development.
+Your job is to generate **production-ready code** for websites and apps.
+Guidelines:
+- Always return three separate files: index.html, styles.css, and script.js
+- HTML must be semantic, responsive, and mobile-first (use <meta viewport>)
+- CSS should use Flexbox/Grid and include hover/transition effects
+- JavaScript must add interactivity (e.g. button actions, animations, toggles)
+- Include a hero section, feature grid, testimonials, and footer
+- Fill with realistic content (no lorem ipsum, no placeholders)
+- Return **only valid JSON** with keys: "files" → { "index.html": "...", "styles.css": "...", "script.js": "..." }
+Prompt: {user_prompt}
+Output:
 """
 # ----------------- CHAINS -----------------
         return {"reply": answer.strip() if isinstance(answer, str) else str(answer)}
     except HfHubHTTPError as e:
         if "exceeded" in str(e).lower() or "quota" in str(e).lower():
+            return {"reply": "⚠️ Daily token limit reached. Try again in 24 hours."}
         raise e
 @app.post("/stt")