adding rag

app.py

@@ -50,10 +50,10 @@ class PDFSearchApp:
         
         if not self.indexed_docs[id]:
             print("Please index documents first")
-            return "Please index documents first"
+            return "Please index documents first", "--"
         if not query:
             print("Please enter a search query")
-            return "Please enter a search query"
+            return "Please enter a search query", "--"
             
         try:
 
@@ -69,10 +69,10 @@ class PDFSearchApp:
 
             print(f"Retrieved image path: {img_path}")
 
-            return img_path
+            return img_path, f"Retrieved page number: {page_num}"
             
         except Exception as e:
-            return f"Error during search: {str(e)}"
+            return f"Error during search: {str(e)}", "--"
 
 def create_ui():
     app = PDFSearchApp()
@@ -108,6 +108,7 @@ def create_ui():
                     label="Number of results"
                 )
                 search_btn = gr.Button("Search")
+                status_message = gr.Textbox(label="Status")
                 results = gr.Image(label="Retrieved Documents")
         
         # Event handlers
@@ -120,7 +121,7 @@ def create_ui():
         search_btn.click(
             fn=app.search_documents,
             inputs=[state, query_input, num_results],
-            outputs=[results]
+            outputs=[results, status_message]
         )
     
     return demo