Update app.py

app.py

@@ -1,38 +1,21 @@
 import os
 import uuid
+import re
 from fastapi import FastAPI, UploadFile, File, Request
-from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
+from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from fastapi.middleware.cors import CORSMiddleware
 from dotenv import load_dotenv
 from pdfminer.high_level import extract_text as pdf_extract_text
 from docx import Document
-from playwright_model import generate_ui_report
-
-# LangChain 1.0.5 for LangGraph / LangSmith
-import langchain
-print("LangChain version:", langchain.__version__)
-
-
-from langgraph.graph import StateGraph, END
-from langsmith import traceable
 import google.generativeai as genai
-
-# --- LangSmith / LangChain Tracing Check ---
-api_key = os.getenv("LANGSMITH_API_KEY")
-print("🔍 LangSmith tracing enabled:", os.getenv("LANGCHAIN_TRACING_V2"))
-print("🔍 LangSmith project:", os.getenv("LANGCHAIN_PROJECT"))
-print("🔍 LangSmith API key detected:", bool(api_key))
-print("🔍 LangSmith workspace ID:", workspace_id if workspace_id else "None")
-
+import fpdf
 
 # ==========================================================
 # 🔧 Setup
 # ==========================================================
-
 load_dotenv("settings.env")
-
 app = FastAPI(title="PowerApps Mockup Generator")
 
 # Enable CORS
@@ -44,19 +27,26 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Directory setup
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 UPLOAD_FOLDER = os.path.join(BASE_DIR, "static", "uploads")
 OUTPUT_FOLDER = os.path.join(BASE_DIR, "static", "outputs")
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 
+# Serve static files
 app.mount("/static", StaticFiles(directory="static"), name="static")
+
+# Setup templates
 templates = Jinja2Templates(directory="templates")
 
+# Gemini setup
+genai.configure(api_key=os.environ.get("GEMINI_API_KEY2"))
+model = genai.GenerativeModel("gemini-2.5-pro")
+
 # ==========================================================
-# 📄 Helper: File Extraction
+# 📄 Helper Function
 # ==========================================================
-
 def extract_text_from_file(filepath: str) -> str:
     """Extract text from a PDF or DOCX file."""
     if filepath.endswith(".pdf"):
@@ -67,49 +57,38 @@ def extract_text_from_file(filepath: str) -> str:
     return ""
 
 # ==========================================================
-# 🤖 LangGraph Agent Setup
+# 🧭 Routes
 # ==========================================================
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    """Serve the main UI."""
+    return templates.TemplateResponse("index.html", {"request": request})
+
+@app.post("/upload")
+async def upload_files(request: Request, files: list[UploadFile] = File(...)):
+    """Handle uploaded files and generate PowerApps-style HTML."""
+    all_text = ""
+    for file in files:
+        filename = f"{uuid.uuid4()}_{file.filename}"
+        filepath = os.path.join(UPLOAD_FOLDER, filename)
+        with open(filepath, "wb") as f:
+            f.write(await file.read())
+        extracted_text = extract_text_from_file(filepath)
+        all_text += f"\n\nDocument: {file.filename}\n{extracted_text}"
 
-GEMINI_KEYS = {
-    "parser": os.getenv("GEMINI_API_KEY"),
-    "requirements": os.getenv("GEMINI_API_KEY1"),
-    "ui": os.getenv("GEMINI_API_KEY2"),
-}
-
-# Configure API keys globally
-genai.configure(api_key=GEMINI_KEYS["parser"])  # parser
-parser_model = genai.GenerativeModel("gemini-2.5-pro")
-
-genai.configure(api_key=GEMINI_KEYS["requirements"])  # requirements
-req_model = genai.GenerativeModel("gemini-2.5-pro")
-
-genai.configure(api_key=GEMINI_KEYS["ui"])  # UI generator
-ui_model = genai.GenerativeModel("gemini-2.5-pro")
-
-class AgentState(dict):
-    """Simple state container for passing data between agents."""
-    pass
-
-@traceable(name="DocumentParserAgent")
-def document_parser_agent(state: AgentState) -> AgentState:
-    text = state.get("input_text", "")
-    result = parser_model.generate_content(f"Extract structured sections and summaries from:\n\n{text}")
-    state["parsed_doc"] = result.text.strip()
-    return state
-
-@traceable(name="RequirementsExtractionAgent")
-def requirements_agent(state: AgentState) -> AgentState:
-    parsed = state.get("parsed_doc", "")
-    result = req_model.generate_content(f"Extract key functional and UI requirements for a recruitment system from this document:\n\n{parsed}")
-    state["requirements"] = result.text.strip()
-    return state
-
-@traceable(name="UIGeneratorAgent")
-def ui_generator_agent(state: AgentState) -> AgentState:
-    requirements = state.get("requirements", "")
-    reference_html = state.get("reference_html", "")
-    all_text = state.get("input_text", "")
+    if not all_text.strip():
+        return JSONResponse({"error": "No readable content found in files."}, status_code=400)
 
+    reference_path = os.path.join(BASE_DIR, "templates", "demo_qms_design.html")
+    if not os.path.exists(reference_path):
+        return JSONResponse({"error": "Reference design not found."}, status_code=500)
+
+    with open(reference_path, "r", encoding="utf-8") as f:
+        reference_html = f.read()
+
+    # ==========================================================
+    # 🧠 Gemini Prompt
+    # ==========================================================
     prompt = f"""
     You are an expert PowerApps-style UI generator and business workflow designer.
     Your goal is to create a Recruitment Management Application HTML layout that follows **exactly the same layout, structure, and JavaScript logic** as the reference QMS app design provided below.
@@ -151,82 +130,55 @@ def ui_generator_agent(state: AgentState) -> AgentState:
     --------------------
     """
 
-    result = ui_model.generate_content(prompt)
-    state["generated_ui"] = result.text.strip()
-    return state
-
-def build_agent_graph():
-    """Build the sequential agent pipeline."""
-    graph = StateGraph(AgentState)
-    graph.add_node("parse", document_parser_agent)
-    graph.add_node("requirements", requirements_agent)
-    graph.add_node("ui", ui_generator_agent)
-
-    graph.add_edge("parse", "requirements")
-    graph.add_edge("requirements", "ui")
-    graph.add_edge("ui", END)
-
-    graph.set_entry_point("parse")
-    return graph.compile()
-
-# ==========================================================
-# 🧭 Routes
-# ==========================================================
-
-@app.get("/", response_class=HTMLResponse)
-async def index(request: Request):
-    return templates.TemplateResponse("index.html", {"request": request})
-
-@app.post("/upload")
-async def upload_files(request: Request, files: list[UploadFile] = File(...)):
-    all_text = ""
-    for file in files:
-        filename = f"{uuid.uuid4()}_{file.filename}"
-        filepath = os.path.join(UPLOAD_FOLDER, filename)
-        with open(filepath, "wb") as f:
-            f.write(await file.read())
-        all_text += extract_text_from_file(filepath) + "\n"
-
-    if not all_text.strip():
-        return JSONResponse({"error": "No readable content found in uploaded files."}, status_code=400)
-
-    reference_path = os.path.join(BASE_DIR, "templates", "demo_qms_design.html")
-    if not os.path.exists(reference_path):
-        return JSONResponse({"error": "Reference design not found."}, status_code=500)
-
-    with open(reference_path, "r", encoding="utf-8") as f:
-        reference_html = f.read()
-
+    # ==========================================================
+    # 🤖 Gemini Call
+    # ==========================================================
     try:
-        graph = build_agent_graph()
-        initial_state = AgentState({
-            "input_text": all_text,
-            "reference_html": reference_html
-        })
-        final_state = graph.invoke(initial_state)
+        response = model.generate_content(prompt)
+        generated_html = response.text.strip()
     except Exception as e:
-        return JSONResponse({"error": f"Agent pipeline failed: {str(e)}"}, status_code=500)
+        return JSONResponse({"error": f"Gemini API error: {str(e)}"}, status_code=500)
 
-    generated_html = final_state.get("generated_ui", "").strip()
+    # Clean up possible markdown fences
     if generated_html.startswith("```"):
-        generated_html = generated_html.split("```html")[-1].split("```")[-1].strip()
+        generated_html = generated_html.split("```html")[-1]
+        generated_html = generated_html.split("```")[-1].strip()
+
+    matches = re.findall(r"<html.*?</html>", generated_html, re.DOTALL | re.IGNORECASE)
+    if matches:
+        generated_html = matches[0]
 
+    # ==========================================================
+    # 💾 Save output
+    # ==========================================================
     output_filename = f"{uuid.uuid4()}.html"
     output_path = os.path.join(OUTPUT_FOLDER, output_filename)
     with open(output_path, "w", encoding="utf-8") as f:
         f.write(generated_html)
 
+    # ==========================================================
+    # 🌐 Auto-detect base URL
+    # ==========================================================
     host = request.url.hostname
     scheme = request.url.scheme
-    public_url = f"{scheme}://{host}/static/outputs/{output_filename}"
+    base_url = f"{scheme}://{host}"
+    public_url = f"{base_url}/static/outputs/{output_filename}"
 
+    # ==========================================================
+    # ✅ Return result
+    # ==========================================================
     return JSONResponse({
         "html": generated_html,
         "link": public_url
     })
 
+
+from fastapi.responses import FileResponse
+from playwright_model import generate_ui_report
+
 @app.post("/generate_report")
 async def generate_report(request: Request):
+    """Generate a UI walkthrough report PDF using Playwright."""
     data = await request.json()
     app_url = data.get("url")
     if not app_url:
@@ -234,4 +186,5 @@ async def generate_report(request: Request):
 
     output_pdf = os.path.join(OUTPUT_FOLDER, "UI_Report.pdf")
     await generate_ui_report(app_url, output_pdf)
+
     return FileResponse(output_pdf, filename="UI_Report.pdf", media_type="application/pdf")