debugging2

final.py

@@ -89,6 +89,45 @@ def extract_entities(text, nlp):
     entities = [(ent.text, ent.label_) for ent in doc.ents]
     return entities
 
+# def update_knowledge_graph(text, is_real, knowledge_graph, nlp, save=True, push_to_hf=True):
+#     """Update knowledge graph with new information"""
+#     entities = extract_entities(text, nlp)
+#     for entity, entity_type in entities:
+#         if not knowledge_graph.has_node(entity):
+#             knowledge_graph.add_node(
+#                 entity,
+#                 type=entity_type,
+#                 real_count=1 if is_real else 0,
+#                 fake_count=0 if is_real else 1
+#             )
+#         else:
+#             if is_real:
+#                 knowledge_graph.nodes[entity]['real_count'] += 1
+#             else:
+#                 knowledge_graph.nodes[entity]['fake_count'] += 1
+
+#     for i, (entity1, _) in enumerate(entities):
+#         for entity2, _ in entities[i+1:]:
+#             if not knowledge_graph.has_edge(entity1, entity2):
+#                 knowledge_graph.add_edge(
+#                     entity1,
+#                     entity2,
+#                     weight=1,
+#                     is_real=is_real
+#                 )
+#             else:
+#                 knowledge_graph[entity1][entity2]['weight'] += 1
+#     if save:
+#         from save_model import save_knowledge_graph, push_to_huggingface
+#         filepath = save_knowledge_graph(knowledge_graph)
+        
+#         # Push to Hugging Face if requested
+#         if push_to_hf:
+#             repo_id = os.getenv("HF_REPO_ID", "HeheBoi0769/Nexus_NLP_model")
+#             push_to_huggingface(filepath, repo_id)
+    
+#     return knowledge_graph
+
 def update_knowledge_graph(text, is_real, knowledge_graph, nlp, save=True, push_to_hf=True):
     """Update knowledge graph with new information"""
     entities = extract_entities(text, nlp)
@@ -117,6 +156,7 @@ def update_knowledge_graph(text, is_real, knowledge_graph, nlp, save=True, push_
                 )
             else:
                 knowledge_graph[entity1][entity2]['weight'] += 1
+    
     if save:
         from save_model import save_knowledge_graph, push_to_huggingface
         filepath = save_knowledge_graph(knowledge_graph)