skills

app.py

@@ -125,7 +125,9 @@ PROMPT = (
     "Extract the following information from the provided text ONLY "
     "Course Code, Course Name, Credit, Delivery method, Course description, and Topical outline and do not add anything else except the information available in this text. "
 )
-
+PROMPT_SKILLS = (
+    "Extract the Course name and Primary Skills from this text. "
+)
 @app.route("/", methods=["GET"])
 def home():
     return jsonify({"message": "Welcome to the PDF Extraction API. Use the /extract endpoint to extract information."})
@@ -145,7 +147,10 @@ def extract_info():
         pdf_text = extract_text_from_pdf(pdf_url)
         prompt = f"{PROMPT}\n\n{pdf_text}"
         response = predict_text(prompt)
-        return jsonify({"extracted_info": response})
+        if "skills" == True:
+            prompt_skills = f"{PROMPT_SKILLS}\n\n{pdf_text}"
+            response_skills = predict_text(prompt_skills)
+        return jsonify({"extracted_info": response + response_skills})
     except Exception as e:
         return jsonify({"error": str(e)}), 500