improve snippet matching

app.py

@@ -64,24 +64,22 @@ def search(term, limit=10, clean=True, strict=True, abstracts=True):
     )
 
 
-
-
-
 def find_source(text, docs):
     for doc in docs:
-        if text in remove_html(doc[1][0]['snippet']):
-            new_text = text
-            for snip in remove_html(doc[1][0]['snippet']).split('.'):
-                if text in snip:
-                    new_text = snip
-            return {
-                'citation_statement': doc[1][0]['snippet'].replace('<strong class="highlight">', '').replace('</strong>', ''),
-                'text': new_text,
-                'from': doc[1][0]['source'],
-                'supporting': doc[1][0]['target'],
-                'source_title': doc[2],
-                'source_link': f"https://scite.ai/reports/{doc[0]}"
-            }
+        for snippet in doc[1]:
+            if text in remove_html(snippet.get('snippet', '')):
+                new_text = text
+                for snip in remove_html(snippet.get('snippet', '')).split('.'):
+                    if text in snip:
+                        new_text = snip
+                return {
+                    'citation_statement': snippet['snippet'].replace('<strong class="highlight">', '').replace('</strong>', ''),
+                    'text': new_text,
+                    'from': snippet['source'],
+                    'supporting': snippet['target'],
+                    'source_title': doc[2],
+                    'source_link': f"https://scite.ai/reports/{doc[0]}"
+                }
     return None