Spaces:

VGreatVig07
/

Assignment_Recommender

Sleeping

App Files Files Community

VGreatVig07 commited on Apr 25, 2025

Commit

214ecb2

verified ·

1 Parent(s): faed9d9

Upload 8 files

Browse files

Files changed (9) hide show

.gitattributes +1 -0
Dockerfile +12 -0
Static/index.html +86 -0
__pycache__/main.cpython-313.pyc +0 -0
main.py +78 -0
render.yaml +10 -0
requirements.txt +103 -0
shl_combined_assessments.csv +0 -0
shl_vector_index.faiss +3 -0

.gitattributes CHANGED Viewed

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 Backend/shl_vector_index.faiss filter=lfs diff=lfs merge=lfs -text

 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 Backend/shl_vector_index.faiss filter=lfs diff=lfs merge=lfs -text
+shl_vector_index.faiss filter=lfs diff=lfs merge=lfs -text

Dockerfile ADDED Viewed

	@@ -0,0 +1,12 @@

+# Dockerfile
+FROM python:3.10-slim
+WORKDIR /app
+COPY . .
+RUN pip install --no-cache-dir -r requirements.txt
+EXPOSE 9000
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

Static/index.html ADDED Viewed

	@@ -0,0 +1,86 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <title>SHL Assessment Recommender</title>
+  <style>
+    body {
+      font-family: Arial, sans-serif;
+      padding: 20px;
+      background-color: #f4f6f8;
+    }
+    input, button {
+      padding: 10px;
+      font-size: 16px;
+      margin-right: 10px;
+    }
+    button {
+      cursor: pointer;
+    }
+    .results {
+      margin-top: 20px;
+    }
+    .card {
+      border: 1px solid #ccc;
+      padding: 15px;
+      margin: 10px 0;
+      border-radius: 10px;
+      background: #ffffff;
+      box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+    }
+    .card h3 {
+      margin: 0 0 10px;
+    }
+  </style>
+</head>
+<body>
+  <h2>🔍 SHL Assessment Recommender</h2>
+  <input type="text" id="queryInput" placeholder="Enter your job role or requirement..." size="50">
+  <button onclick="sendQuery()">Search</button>
+  <div class="results" id="results"></div>
+  <script>
+    async function sendQuery() {
+      const query = document.getElementById("queryInput").value;
+      const resultsDiv = document.getElementById("results");
+      resultsDiv.innerHTML = "<p>Searching...</p>";
+      try {
+        const response = await fetch("http://localhost:8000/query", {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json"
+          },
+          body: JSON.stringify({ query })
+        });
+        const data = await response.json();
+        if (data.recommendations && data.recommendations.length > 0) {
+          resultsDiv.innerHTML = data.recommendations.map((r, index) => `
+            <div class="card">
+              <h3>🏆 Rank ${index + 1}: ${r.Assignment_Name}</h3>
+              <p>🔗 <a href="${r.Assignment_Link}" target="_blank">${r.Assignment_Link}</a></p>
+              <p>🧪 Test Type: ${r.Test_Type}</p>
+              <p>⏱️ Duration: ${r.Approximate_Completion_Time} mins</p>
+              <p>🌐 Remote Testing: ${r.Remote_Testing_Support}</p>
+              <p>📊 Adaptive/IRT: ${r.Adaptive_IRT_Support}</p>
+              <p>👤 Job Levels: ${r.Job_Levels}</p>
+            </div>
+          `).join("");
+        } else {
+          resultsDiv.innerHTML = "<p>No results found.</p>";
+        }
+      } catch (error) {
+        resultsDiv.innerHTML = `<p style="color:red;">Error: ${error.message}</p>`;
+      }
+    }
+  </script>
+</body>
+</html>

__pycache__/main.cpython-313.pyc ADDED Viewed

Binary file (3.96 kB). View file

main.py ADDED Viewed

	@@ -0,0 +1,78 @@

+from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+from pydantic import BaseModel
+import google.generativeai as genai
+from dotenv import load_dotenv
+from fastapi.middleware.cors import CORSMiddleware
+import os
+import json
+import pandas as pd
+import faiss
+from sentence_transformers import SentenceTransformer
+import uvicorn
+import math
+# Load environment variables and configure Gemini
+load_dotenv()
+genai.configure(api_key=os.getenv("API_KEY"))
+gemini_model = genai.GenerativeModel("gemini-2.0-flash")
+# Load vector index and dataframe
+index = faiss.read_index("shl_vector_index.faiss")
+df = pd.read_csv("shl_combined_assessments.csv")
+model = SentenceTransformer("all-MiniLM-L6-v2")
+# FastAPI app
+app = FastAPI()
+app.mount("/static", StaticFiles(directory="Static"), name="static")
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # or specify your frontend origin like ["http://localhost:3000"]
+    allow_credentials=True,
+    allow_methods=["*"],  # allows all HTTP methods including OPTIONS
+    allow_headers=["*"],
+)
+@app.get("/")
+async def read_index():
+    return FileResponse(os.path.join("Static", "index.html"))
+# Request schema
+class Query(BaseModel):
+    query: str
+# Format result
+def format_row(row):
+    def safe_cast(val, cast_type, default):
+        try:
+            if val is None or (isinstance(val, float) and math.isnan(val)):
+                return default
+            return cast_type(val)
+        except Exception:
+            return default
+    return {
+        "Assignment_Name": str(row["Assignment_Name"]),
+        "Assignment_Link": str(row["Assignment_Link"]),
+        "Test_Type": str(row["Test_Type"]),
+        "Approximate_Completion_Time": safe_cast(row["Approximate_Completion_Time"], int, -1),
+        "Remote_Testing_Support": bool(row["Remote_Testing_Support"]),
+        "Adaptive_IRT_Support": bool(row["Adaptive_IRT_Support"]),
+        "Job_Levels": str(row.get("Job_Levels", "N/A")),
+    }
+# API endpoint
+@app.post("/query")
+async def query_assessments(payload: Query):
+    query_embedding = model.encode([payload.query]).astype("float32")
+    D, I = index.search(query_embedding, 10)  # Top 10 results
+    results = [format_row(df.iloc[idx]) for idx in I[0]]
+    return {"query": payload.query, "recommendations": results}
+# if __name__ == "__main__":
+#     uvicorn.run("main:app", host="0.0.0.0", port=9000, reload=True)

render.yaml ADDED Viewed

	@@ -0,0 +1,10 @@

+services:
+  - type: web
+    name: shl-fastapi-backend
+    runtime: python
+    buildCommand: pip install -r requirements.txt
+    startCommand: uvicorn main:app --host 0.0.0.0 --port 9000
+    envVars:
+      - key: PORT
+        value: 9000
+    plan: free

requirements.txt ADDED Viewed

	@@ -0,0 +1,103 @@

+aiofiles==24.1.0
+annotated-types==0.7.0
+anyio==4.9.0
+attrs==25.3.0
+beautifulsoup4==4.13.3
+bs4==0.0.2
+cachetools==5.5.2
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+faiss-cpu==1.10.0
+fastapi==0.115.12
+filelock==3.18.0
+firecrawl-py==1.15.0
+fsspec==2025.3.2
+google==3.0.0
+google-ai-generativelanguage==0.6.15
+google-api-core==2.24.2
+google-api-python-client==2.166.0
+google-auth==2.38.0
+google-auth-httplib2==0.2.0
+google-genai==1.9.0
+google-generativeai==0.8.4
+google.py==0.1.1
+googleapis-common-protos==1.69.2
+greenlet==3.1.1
+grpcio==1.71.0
+grpcio-status==1.71.0
+h11==0.14.0
+httpcore==1.0.7
+httplib2==0.22.0
+httpx==0.28.1
+huggingface-hub==0.30.1
+idna==3.10
+Jinja2==3.1.6
+joblib==1.4.2
+jsonpatch==1.33
+jsonpointer==3.0.0
+langchain==0.3.23
+langchain-core==0.3.51
+langchain-text-splitters==0.3.8
+langsmith==0.3.24
+MarkupSafe==3.0.2
+mpmath==1.3.0
+nest-asyncio==1.6.0
+networkx==3.4.2
+numpy==2.2.4
+orjson==3.10.16
+outcome==1.3.0.post0
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+playwright==1.51.0
+proto-plus==1.26.1
+protobuf==5.29.4
+pyasn1==0.6.1
+pyasn1_modules==0.4.2
+pydantic==2.11.2
+pydantic_core==2.33.1
+pyee==12.1.1
+pyparsing==3.2.3
+PySocks==1.7.1
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+requests-toolbelt==1.0.0
+rsa==4.9
+safetensors==0.5.3
+scikit-learn==1.6.1
+scipy==1.15.2
+selenium==4.31.0
+sentence-transformers==4.0.2
+setuptools==78.1.0
+six==1.17.0
+sniffio==1.3.1
+sortedcontainers==2.4.0
+soupsieve==2.6
+SQLAlchemy==2.0.40
+starlette==0.46.1
+sympy==1.13.1
+tenacity==9.1.2
+threadpool==1.3.2
+threadpoolctl==3.6.0
+tokenizers==0.21.1
+torch==2.6.0
+tqdm==4.67.1
+transformers==4.50.3
+trio==0.29.0
+trio-websocket==0.12.2
+typing-inspection==0.4.0
+typing_extensions==4.13.1
+tzdata==2025.2
+uritemplate==4.1.1
+urllib3==2.3.0
+uvicorn==0.34.0
+webdriver-manager==4.0.2
+websocket-client==1.8.0
+websockets==15.0.1
+wsproto==1.2.0
+zstandard==0.23.0

shl_combined_assessments.csv ADDED Viewed

The diff for this file is too large to render. See raw diff

shl_vector_index.faiss ADDED Viewed

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

+version https://git-lfs.github.com/spec/v1
+oid sha256:56ece2437be1d6a57dab2687c453c864f3c7ac5445a73e334f653e25d2a6e0cd
+size 795693