Upload PromptForge v1.0 — Structured prompt generator for Google AI Studio

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "github.copilot.chat.codeGeneration.useInstructionFiles": true
+}

backend/instruction_store.py

@@ -9,7 +9,7 @@ from datetime import datetime
 from pathlib import Path
 from typing import Dict, List, Optional
 
-from schemas import (
+from .schemas import (
     InstructionSettings, InstructionSettingsCreate, InstructionSettingsUpdate,
     AIProvider, OutputFormat, PersonaType, StyleType,
 )

backend/main.py

@@ -7,21 +7,21 @@ import logging, os
 from pathlib import Path
 from typing import List, Optional
 
-from fastapi import FastAPI, HTTPException, Query, status
+from fastapi import FastAPI, HTTPException, Query, status, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 
-import store
-import instruction_store
-from ai_client import enhance_prompt, check_hf_key, check_google_key, HF_DEFAULT_MODEL, GOOGLE_DEFAULT_MODEL
-from prompt_logic import (
+from . import store
+from . import instruction_store
+from .ai_client import enhance_prompt, check_hf_key, check_google_key, HF_DEFAULT_MODEL, GOOGLE_DEFAULT_MODEL
+from .prompt_logic import (
     build_manifest, build_manifest_from_settings,
     apply_edits, refine_with_feedback, generate_explanation,
     apply_target_formatting,
 )
-from schemas import (
+from .schemas import (
     ApproveRequest, ApproveResponse,
     ExportRequest, ExportResponse,
     GenerateRequest, GenerateFromSettingsRequest, GenerateResponse,
@@ -109,6 +109,28 @@ async def get_config() -> EnvConfigStatus:
     )
 
 
+@app.post("/api/upload-main", tags=["System"])
+async def upload_main(file: UploadFile = File(...)) -> dict:
+    """Upload an HTML file and replace the frontend index.html (simple uploader).
+    The uploaded file must be an HTML file. The previous index.html is backed up
+    to `index.html.bak` if present.
+    """
+    if not file.filename.lower().endswith(".html"):
+        raise HTTPException(status_code=400, detail="Only .html files are allowed.")
+    try:
+        content = await file.read()
+        dest = _FRONTEND_DIR / "index.html"
+        # Backup existing index
+        if dest.exists():
+            bak = _FRONTEND_DIR / "index.html.bak"
+            bak.write_bytes(dest.read_bytes())
+        dest.write_bytes(content)
+    except Exception as exc:
+        logger.exception("Failed to save uploaded main file")
+        raise HTTPException(status_code=500, detail=str(exc))
+    return {"success": True, "message": "Uploaded as main index.", "link": "/"}
+
+
 @app.get("/api/stats", response_model=StatsResponse, tags=["System"])
 async def get_stats() -> StatsResponse:
     s = store.get_stats()

backend/prompt_logic.py

@@ -9,7 +9,7 @@ import textwrap
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Tuple
 
-from schemas import (
+from .schemas import (
     PromptManifest,
     StructuredPrompt,
     PersonaType,

backend/schemas.py

@@ -298,9 +298,4 @@ class EnvConfigStatus(BaseModel):
     port:           str  = "7860"
     version:        str  = "4.0"
     hf_model:       str  = "mistralai/Mistral-7B-Instruct-v0.2"
-    google_model:   str  = "gemini-1.5-flash"
-class TargetModel(str, Enum):
-    google_ai_studio = "google_ai_studio"
-    openai = "openai"
-    anthropic = "anthropic"
-    generic = "generic"
+    google_model:   str  = "gemini-1.5-flash"

backend/store.py

@@ -10,7 +10,7 @@ from datetime import datetime
 from pathlib import Path
 from typing import Dict, List, Optional
 
-from schemas import PromptManifest, HistoryEntry, PromptStatus
+from .schemas import PromptManifest, HistoryEntry, PromptStatus
 
 logger = logging.getLogger("promptforge.store")