Update inference.py

inference.py

@@ -4,6 +4,7 @@ import os
 import threading 
 import base64
 import os
+import re
 import threading
 from google import genai
 from google.genai import types
@@ -39,11 +40,11 @@ def feedback(query, context, history):
     msg["From"] = mail
     msg["To"] = mail
     try: 
-        with smtplib.SMTP_SSL("smtp.gmail.com", 2525, context=ssl.create_default_context()) as server:
+        with smtplib.SMTP_SSL("smtp.gmail.com", 465, context=ssl.create_default_context()) as server:
             server.login(mail, APP_PASSWORD)
             server.send_message(msg)
-    except Exception as e: 
-       print(f'envio fallido {query}, error: {e}')
+    except: 
+       print(f'envio fallido {query}')
 
 def query_db(query, limit = 5, filters = {}, measure = "cosine_distance", include_value = True, include_metadata=True, table = "2023"):
   query_embeds = vx.get_or_create_collection(name= table, dimension=384)
@@ -187,12 +188,44 @@ def generate(query, context, message_history):
     )
     return response.text
 
-def inference(query, history, context):
+def generate_links(judgments):
+    output = []
+    
+    for judgment in judgments:
+        # Match patterns like 'C-186-19', 'T-092-93', 'SU-000-20'
+        match = re.match(r'([A-Za-z]+)-(\d{3})-(\d{2})', judgment)
+        if match:
+            type_code = match.group(1).lower()  # 'c', 't', 'su'
+            number = match.group(2)
+            year_suffix = match.group(3)
+            
+            # Determine full 4-digit year
+            year = '20' + year_suffix if int(year_suffix) <= 68 else '19' + year_suffix
+            
+            # Format radicado for URL
+            if type_code == 'su':
+                radicado_url = f"su{number.zfill(3)}-{year_suffix}"
+            else:
+                radicado_url = f"{type_code}-{number}-{year_suffix}"  # note the dash after type
+            
+            # Build URL
+            url = f"https://www.corteconstitucional.gov.co/relatoria/{year}/{radicado_url}.htm"
+            
+            # Append as a single sublist with judgment and url
+            output.append([[judgment], [url]])
+        else:
+            print(f"Invalid format: {judgment}")
+    
+    return output
+
+def link_constructor(judgements):
+   return generate_links([x.rsplit('.rtf')[0] for x in judgements])
+
+def inference(query, history, context, links):
   if context == None or len(context) <= 0 or len(history) <= 0:
     vector_query_results = vector_query(query)
     context = context_draft(vector_query_results, query)
     t = threading.Thread(target=feedback, args=(query, context, history))
-    t.start()
-    return generate(query, context, history), context
+    return generate(query, context, history), context, link_constructor(vector_query_results)
   else:
-    return generate(query, context, history), context
+    return generate(query, context, history), context, links