Krish Patel commited on
Commit
0e5ed12
·
1 Parent(s): 25bb7ca

Debugging1

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -42,7 +42,7 @@ def generate_knowledge_graph_viz(text, nlp, tokenizer, model):
42
  # Update knowledge graph
43
  kg_builder.update_knowledge_graph(text, not is_fake, nlp)
44
 
45
- # Randomly select subset of edges (e.g. 60% of edges)
46
  edges = list(kg_builder.knowledge_graph.edges())
47
  selected_edges = random.sample(edges, k=int(len(edges) * 0.1))
48
 
@@ -154,10 +154,14 @@ def main():
154
  if gemini_result and gemini_result.get('gemini_analysis'):
155
  break
156
 
157
- except Exception:
158
- pass
 
159
 
160
  retry_count += 1
 
 
 
161
 
162
  # Use default values if all retries failed
163
  if not gemini_result:
 
42
  # Update knowledge graph
43
  kg_builder.update_knowledge_graph(text, not is_fake, nlp)
44
 
45
+ # Randomly select subset of edges (e.g. 10% of edges)
46
  edges = list(kg_builder.knowledge_graph.edges())
47
  selected_edges = random.sample(edges, k=int(len(edges) * 0.1))
48
 
 
154
  if gemini_result and gemini_result.get('gemini_analysis'):
155
  break
156
 
157
+ except Exception as e:
158
+ st.error(f"Gemini API error: {str(e)}")
159
+ print(f"Gemini error: {str(e)}")
160
 
161
  retry_count += 1
162
+ import time
163
+ time.sleep(1) # Add a 1-second delay between retries
164
+
165
 
166
  # Use default values if all retries failed
167
  if not gemini_result: