Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from dotenv import load_dotenv
+import os
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 from llama_index.llama_pack.base import BaseLlamaPack
@@ -10,7 +11,6 @@ from llama_index import ServiceContext
 from llama_index.schema import NodeWithScore
 from llama_index.response_synthesizers import TreeSummarize
 from pydantic import BaseModel
-import os
 import pdfplumber
 import io
 
@@ -49,19 +49,22 @@ def _format_criteria_str(criteria: List[str]) -> str:
     return criteria_str
 
 class ResumeScreenerPack(BaseLlamaPack):
-    def _init_(
-        self, job_description: str, criteria: List[str], llm: Optional[LLM] = None
+    def __init__(
+        self,
+        job_description: str = "",
+        criteria: List[str] = [],
+        llm: Optional[LLM] = None
     ) -> None:
         self.reader = PDFReader()
         llm = llm or OpenAI(model="gpt-4", api_key=openai_api_key)
         service_context = ServiceContext.from_defaults(llm=llm)
-        self.synthesizer = TreeSummarize(
-            output_cls=ResumeScreenerDecision, service_context=service_context
-        )
         criteria_str = _format_criteria_str(criteria)
         self.query = QUERY_TEMPLATE.format(
             job_description=job_description, criteria_str=criteria_str
         )
+        self.synthesizer = TreeSummarize(
+            output_cls=ResumeScreenerDecision, service_context=service_context
+        )
 
     def get_modules(self) -> Dict[str, Any]:
         """Get modules."""
@@ -116,4 +119,4 @@ def extract_text_from_pdf(uploaded_file):
         return ""
 
 if __name__ == "__main__":
-    main()
+    main()