Update app.py

app.py

@@ -1,7 +1,8 @@
 import os
-import gradio as gr
+import atexit
 import weaviate
 import requests
+import gradio as gr
 from weaviate.classes.init import Auth
 from weaviate.classes.query import Rerank
 from transformers import AutoTokenizer, pipeline, AutoModelForSeq2SeqLM
@@ -29,6 +30,8 @@ def connect_to_db():
     )
     return client
 
+client = connect_to_db()
+
 # ========== تحميل الموديلات ==========
 model_checkpoint = "EN3IMI/RouterAraBERT"
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
@@ -156,7 +159,6 @@ def llm_response_laws(query, docs):
 
 # ========== النظام الرئيسي ==========
 def rag_system(user_input):
-    client = connect_to_db()
     queries, faq_docs = search_for_faq(user_input, client)
     use_laws = router_decision(queries)
 
@@ -166,10 +168,16 @@ def rag_system(user_input):
         law_docs = search_for_laws(user_input, client)
         answer = llm_response_laws(user_input, law_docs)
         
-
-    client.close()
     return answer
 
+@atexit.register
+def close_client():
+    try:
+        client.close()
+        print("Weaviate client closed cleanly.")
+    except Exception as e:
+        print(f"Error closing client: {e}")
+
 # ========== واجهة Gradio ==========
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 AILS RAG System")