Update app.py

app.py

@@ -88,39 +88,39 @@ except Exception as e:
 ## Let's a do simple test from the doc --
 
 # Define query and sources
-#query = "What is the capital of France?"
-#log_debug(f"🔍 Test Query: {query}")
-
-#sources = [
-#    {
-#        "text": "Paris is the capital and most populous city of France.",
-#        "metadata": {"source": "Geographic Encyclopedia", "reliability": "high"}
-#    },
-#    {
-#        "text": "The Eiffel Tower is located in Paris, France.",
-#        "metadata": {"source": "Travel Guide", "year": 2020}
-#    }
-#]
-#log_debug("📄 Test Sources loaded successfully.")
+query = "What is the capital of France?"
+log_debug(f"🔍 Test Query: {query}")
+
+sources = [
+    {
+        "text": "Paris is the capital and most populous city of France.",
+        "metadata": {"source": "Geographic Encyclopedia", "reliability": "high"}
+    },
+    {
+        "text": "The Eiffel Tower is located in Paris, France.",
+        "metadata": {"source": "Travel Guide", "year": 2020}
+    }
+]
+log_debug("📄 Test Sources loaded successfully.")
 
 # Generate a response
-#try:
-#    log_debug("🧠 Test rag model on simple example...")
+try:
+    log_debug("🧠 Test rag model on simple example...")
 #    rag1 = RAGWithCitations(
 #      model_path_or_name="PleIAs/Pleias-RAG-350M"
 #    )
-#    response = rag1.generate(query, 
-#                            sources #,
+    response = rag.generate(query, 
+                            sources #,
                            # do_sample=True,  # Enable sampling
                            # top_p=0.95,     # Set top_p for nucleus sampling
                            # pad_token_id=rag.tokenizer.eos_token_id,  # Set pad_token_id to eos_token_id
                            # attention_mask=None  # Ensure attention_mask is passed if needed
-#                           )
-#    log_debug("✅ Test Answer generated successfully.")
-#    log_debug(response["processed"]["clean_answer"])
-#except Exception as e:
-#    log_debug(f"❌ Test Answer generation failed: {str(e)}")
-#    raise
+                           )
+    log_debug("✅ Test Answer generated successfully.")
+    log_debug(response["processed"]["clean_answer"])
+except Exception as e:
+    log_debug(f"❌ Test Answer generation failed: {str(e)}")
+    raise