Added backend files

README.md

@@ -1,10 +1,19 @@
----
-title: Hackathon Idea Generator
-emoji: 🔥
-colorFrom: pink
-colorTo: gray
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Hackathon Idea Generator API (FastAPI + GPT-2)
+
+This is a backend that uses GPT-2 (hosted on Hugging Face Inference API) to generate:
+
+✔ 5 hackathon ideas  
+✔ Overview  
+✔ Why it's useful  
+✔ Difficulty  
+✔ "How to build it" guide  
+✔ Best pick  
+
+## Endpoints
+
+### `/generate` (POST)
+Request:
+```json
+{
+  "topic": "AI in healthcare"
+}

app.py

@@ -0,0 +1,68 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from model_loader import tokenizer, model
+from prompt_templates import MAIN_PROMPT_TEMPLATE, DETAIL_PROMPT_TEMPLATE
+import torch
+import json
+import re
+
+app = FastAPI()
+
+class GenerateRequest(BaseModel):
+    custom_prompt: str = ""
+
+class DetailRequest(BaseModel):
+    idea_id: int
+    idea_title: str
+
+
+@app.get("/")
+def home():
+    return {"status": "Local GPT-2 Backend Running", "endpoints": ["/generate", "/details"]}
+
+
+def run_gpt2(prompt: str):
+    inputs = tokenizer.encode(prompt, return_tensors="pt")
+    outputs = model.generate(
+        inputs,
+        max_new_tokens=500,
+        temperature=0.8,
+        top_p=0.9,
+        do_sample=True
+    )
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+
+def extract_json(text):
+    try:
+        match = re.search(r"\{[\s\S]*\}", text)
+        if match:
+            return json.loads(match.group())
+    except:
+        pass
+    return {"error": "JSON parsing failed", "raw_output": text}
+
+
+# -----------------------
+# STEP 1 → GENERATE IDEAS
+# -----------------------
+@app.post("/generate")
+def generate(req: GenerateRequest):
+    prompt = MAIN_PROMPT_TEMPLATE.format(CUSTOM_PROMPT=req.custom_prompt)
+    raw_output = run_gpt2(prompt)
+    json_data = extract_json(raw_output)
+    return json_data
+
+
+# -----------------------
+# STEP 2 → IDEA DETAILS
+# -----------------------
+@app.post("/details")
+def details(req: DetailRequest):
+    prompt = DETAIL_PROMPT_TEMPLATE.format(
+        IDEA_ID=req.idea_id,
+        IDEA_TITLE=req.idea_title
+    )
+    raw_output = run_gpt2(prompt)
+    json_data = extract_json(raw_output)
+    return json_data

model_loader.py

@@ -0,0 +1,9 @@
+from transformers import GPT2Tokenizer, GPT2LMHeadModel
+
+def load_model():
+    print("Loading GPT-2 model...")
+    tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+    model = GPT2LMHeadModel.from_pretrained("gpt2")
+    return tokenizer, model
+
+tokenizer, model = load_model()

prompt_templates.py

@@ -0,0 +1,72 @@
+MAIN_PROMPT_TEMPLATE = """
+Generate 3 distinct client-side hackathon ideas. {CUSTOM_PROMPT}
+For each idea, provide a title, a short elevator pitch, an overview, primary tech stack, difficulty, and estimated time.
+Pick the best one and explain why.
+Return ONLY valid JSON matching this exact schema:
+
+{
+  "ideas": [
+    {
+      "id": <int>,
+      "title": "<string>",
+      "elevator": "<string>",
+      "overview": "<string>",
+      "primary_tech_stack": ["<string>", "..."],
+      "difficulty": "<Easy|Medium|Hard>",
+      "time_estimate_hours": <number>
+    }
+  ],
+  "best_pick_id": <int>,
+  "best_pick_reason": "<string>"
+}
+
+Constraints:
+- Return exactly 3 ideas.
+- Keep all text concise for UI cards.
+""".strip()
+
+
+DETAIL_PROMPT_TEMPLATE = """
+Expand on the hackathon idea "{IDEA_TITLE}" (ID: {IDEA_ID}).
+Create a detailed, step-by-step build plan suitable for a 48-hour hackathon.
+Return ONLY valid JSON matching this exact schema:
+
+{
+  "id": {IDEA_ID},
+  "title": "{IDEA_TITLE}",
+  "mermaid_architecture": "<string starting with 'graph LR'>",
+  "phases": [
+    {
+      "name": "<MVP|Polish|Demo>",
+      "time_hours": <number>,
+      "tasks": ["<string>", "..."],
+      "deliverables": ["<string>", "..."]
+    }
+  ],
+  "critical_code_snippets": [
+    {
+      "title": "<string>",
+      "language": "javascript|html|css",
+      "code": "<multiline string of code>"
+    }
+  ],
+  "ui_components": [
+    {
+      "name": "<string>",
+      "purpose": "<string>"
+    }
+  ],
+  "risks_and_mitigations": [
+    {
+      "risk": "<string>",
+      "mitigation":"<string>"
+    }
+  ]
+}
+
+Constraints:
+- The 'phases' array must cover the entire build process.
+- Total time for all phases must not exceed 48 hours.
+- Mermaid diagrams must be valid and represent the project architecture.
+- Include at least one meaningful code snippet.
+""".strip()

requirements.txt

@@ -0,0 +1,5 @@
+fastapi
+uvicorn
+transformers
+torch
+python-multipart