dwb2023 commited on
Commit
e29fb0f
·
verified ·
1 Parent(s): 5ac3adf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -7
app.py CHANGED
@@ -274,19 +274,24 @@ class EventAnalyzer:
274
  ))
275
 
276
  # Adjust confidence based on entity frequency
277
- cursor = self.relationship_engine.conn.execute('''
 
 
 
 
 
 
 
278
  SELECT AVG(frequency) as avg_freq
279
  FROM entities
280
  WHERE entity_text IN (
281
  SELECT DISTINCT entity_text
282
  FROM entities
283
- WHERE entity_text IN ({})
284
  )
285
- '''.format(','.join('?' * len([
286
- *entities["people"],
287
- *entities["organizations"],
288
- *entities["locations"]
289
- ]))))
290
 
291
  avg_frequency = cursor.fetchone()[0] or 1
292
  frequency_boost = min(0.2, (avg_frequency - 1) * 0.05) # Max 0.2 boost for frequency
 
274
  ))
275
 
276
  # Adjust confidence based on entity frequency
277
+ entity_params = [
278
+ *entities["people"],
279
+ *entities["organizations"],
280
+ *entities["locations"]
281
+ ]
282
+
283
+ cursor = self.relationship_engine.conn.execute(
284
+ f'''
285
  SELECT AVG(frequency) as avg_freq
286
  FROM entities
287
  WHERE entity_text IN (
288
  SELECT DISTINCT entity_text
289
  FROM entities
290
+ WHERE entity_text IN ({','.join(['?']*len(entity_params))})
291
  )
292
+ ''',
293
+ entity_params # Pass parameters here
294
+ )
 
 
295
 
296
  avg_frequency = cursor.fetchone()[0] or 1
297
  frequency_boost = min(0.2, (avg_frequency - 1) * 0.05) # Max 0.2 boost for frequency