Fixed app v2

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from llama_cpp import Llama
+import json
 
-# Load the SecurityLLM model
+# Load the Llama model
 try:
     llm = Llama.from_pretrained(
-        repo_id="QuantFactory/SecurityLLM-GGUF",
-        filename="SecurityLLM.Q5_K_M.gguf",  # Ensure the file path is correct
+        repo_id="HuggingFaceTB/SmolLM2-360M-Instruct-GGUF",
+        filename="smollm2-360m-instruct-q8_0.gguf"  # Replace with the correct path to your GGUF file
     )
 except Exception as e:
     raise RuntimeError(f"Failed to load model: {e}")
@@ -33,25 +34,34 @@ def match_cv_to_jobs(cv_text, job_descriptions):
             ]
         )
         
-        # Extract and store the analysis text
+        # Extract the analysis text
         analysis_text = response["choices"][0]["message"]["content"]
-        results.append({
-            "Job Description": description,
-            "Analysis": analysis_text
-        })
+        
+        # Attempt to parse as JSON; if not JSON, use the raw text
+        try:
+            response_data = json.loads(analysis_text)
+            results.append(response_data)
+        except json.JSONDecodeError:
+            results.append({
+                "Job Description": description,
+                "Analysis": analysis_text  # Use raw response if JSON parsing fails
+            })
     
     return results
 
 # Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# CV to Job Description Matcher")
+    gr.Markdown("# CV and Job Description Matcher")
     
+    # Input fields for CV and job descriptions
     cv_text = gr.Textbox(label="CV Text", placeholder="Enter the CV text here", lines=10)
     job_descriptions = gr.Textbox(label="Job Descriptions (one per line)", placeholder="Enter each job description on a new line", lines=5)
-    match_button = gr.Button("Match CV to Job Descriptions")
     
+    # Button and output area
+    match_button = gr.Button("Match CV to Job Descriptions")
     output = gr.JSON(label="Match Results")
     
+    # Set button click to run the function
     match_button.click(fn=match_cv_to_jobs, inputs=[cv_text, job_descriptions], outputs=output)
 
 demo.launch()