commit

REC.py

@@ -21,13 +21,4 @@ def recommend_clause(clause, agreements= agreements.loc[:2], threshold=0.4):
     if distances[idx] < threshold:
         indices_ini.append(idx)
         indices_ini = list(set(indices_ini))
-    return indices_ini
-
-def print_agreements():
-  if loan == 'O':
-    indices.append(3)
-  if insurance == 'O':
-    indices.append(4)
-    indices.append(5)
-  result = agreements.loc[indices]
-  return result
+    return indices_ini

app.py

@@ -207,6 +207,15 @@ def extract_ocr_text(ocr_result):
             indices = recommend_clause(clause = clauses)
             loan = 'O'
             insurance = 'O'
+
+            def print_agreements():
+                if loan == 'O':
+                    indices.append(3)
+                if insurance == 'O':
+                    indices.append(4)
+                    indices.append(5)
+                result = agreements.loc[indices]
+                return result
             rec_df = print_agreements()
 
             for idx, row in rec_df.iterrows():