Update app.py

app.py

@@ -43,7 +43,24 @@ if uploaded_files1:
             text_data = ""
             for page in pdf_reader.pages:
                 text_data += page.extract_text()
-            all_resumes_text1.append(text_data)
+                model = GLiNER.from_pretrained("urchade/gliner_base")
+                labels = ["person", "country", "organization", "time", "role"]
+                entities = model.predict_entities(text_data, labels)
+
+            entity_dict = {}
+            for label in labels:
+                entity_dict[label] = [entity["text"] for entity in entities if entity["label"] == label]
+
+            data = {"Text": text_data, **entity_dict}
+           
+
+
+
+
+
+
+                
+            all_resumes_text1.append(data)
         except Exception as e:
             st.error(f"Error processing file {uploaded_file.name}: {e}")
 
@@ -86,6 +103,19 @@ if uploaded_files2:
             text_data = ""
             for page in pdf_reader.pages:
                 text_data += page.extract_text()
+                model = GLiNER.from_pretrained("urchade/gliner_base")
+                labels = ["person", "country", "organization", "time", "role"]
+                entities = model.predict_entities(text_data, labels)
+
+            entity_dict = {}
+            for label in labels:
+                entity_dict[label] = [entity["text"] for entity in entities if entity["label"] == label]
+
+            data = {"Text": text_data, **entity_dict}
+           
+
+                
+            
             all_resumes_text2.append(text_data)
         except Exception as e:
             st.error(f"Error processing file {uploaded_file.name}: {e}")