Update rag_system.py

rag_system.py

@@ -35,8 +35,14 @@ class RAGPipeline:
             n_results=3
         )
 
-        print("🔍 Top Documents:")
+        print("\n🔍 Top Documents:")
         for i, doc in enumerate(results["documents"][0]):
-            print(f"{i+1}. {doc[:200]}...
-")
-        return results["documents"][0][0]
+            print(f"{i+1}. {doc[:200]}...\n")
+
+        # HF Spaces output üçün cavabı qaytar
+        return "\n\n".join([f"{i+1}. {doc[:500]}" for i, doc in enumerate(results["documents"][0])])
+
+if __name__ == "__main__":
+    rag = RAGPipeline()
+    rag.index_document("sample.pdf")
+    rag.query("What is this document about?")