Update app.py

app.py

@@ -275,9 +275,9 @@ async def process_query(request: QueryRequest):
             if language_code == 0:
                 query_text = translate_text(query_text, 'ar_to_en')
 
-            query_embedding = models['embedding'].encode([query_text])
+            query_embedding = models['embedding'].encode([query_text])  # Ensure the model has been loaded and is available.
             relevant_docs = query_embeddings(query_embedding)
-            
+
             if not relevant_docs:
                 return {
                     'answer': 'No relevant information found. Please try a different query.',
@@ -286,7 +286,7 @@ async def process_query(request: QueryRequest):
 
             doc_texts = [retrieve_document_text(doc_id) for doc_id, _ in relevant_docs]
             doc_texts = [text for text in doc_texts if text.strip()]
-            
+
             if not doc_texts:
                 return {
                     'answer': 'Unable to retrieve relevant documents. Please try again.',
@@ -295,15 +295,16 @@ async def process_query(request: QueryRequest):
 
             rerank_scores = rerank_documents(query_text, doc_texts)
             ranked_texts = [text for _, text in sorted(zip(rerank_scores, doc_texts), reverse=True)]
-            
+
             context = " ".join(ranked_texts[:3])
             answer = generate_answer(query_text, context)
-            
+
             if language_code == 0:
                 answer = translate_text(answer, 'en_to_ar')
 
             return {
                 'answer': answer,
+                'reranked_documents': ranked_texts,
                 'success': True
             }
 
@@ -323,7 +324,7 @@ async def process_query(request: QueryRequest):
 
 # Initialize application
 print("Initializing application...")
-init_success = init_nltk() and load_models() and load_data()
+init_success = load_models() and load_data()
 
 if not init_success:
     print("Warning: Application initialized with partial functionality")