adding rag

app.py

@@ -6,6 +6,9 @@ import uuid
 
 
 from middleware import Middleware
+from rag import Rag
+
+rag = Rag()
 
 def generate_uuid(state):
     # Check if UUID already exists in session state
@@ -44,7 +47,7 @@ class PDFSearchApp:
             return f"Error processing PDF: {str(e)}"
     
     
-    def search_documents(self, state, query, num_results=5):
+    def search_documents(self, state, query, num_results=1):
         print(f"Searching for query: {query}")
         id = generate_uuid(state)
         
@@ -69,7 +72,9 @@ class PDFSearchApp:
 
             print(f"Retrieved image path: {img_path}")
 
-            return img_path, f"Retrieved page number: {page_num}"
+            rag_response = rag.get_answer_from_gemini(query, [img_path])
+
+            return img_path, rag_response
             
         except Exception as e:
             return f"Error during search: {str(e)}", "--"
@@ -100,16 +105,16 @@ def create_ui():
         with gr.Tab("Query"):
             with gr.Column():
                 query_input = gr.Textbox(label="Enter query")
-                num_results = gr.Slider(
-                    minimum=1,
-                    maximum=10,
-                    value=5,
-                    step=1,
-                    label="Number of results"
-                )
+                # num_results = gr.Slider(
+                #     minimum=1,
+                #     maximum=10,
+                #     value=5,
+                #     step=1,
+                #     label="Number of results"
+                # )
                 search_btn = gr.Button("Query")
-                llm_answer = gr.Textbox(label="LLM Answer", interactive=False)
-                images = gr.Image(label="Pages used for RAG")
+                llm_answer = gr.Textbox(label="RAG Response", interactive=False)
+                images = gr.Image(label="Page matching query")
         
         # Event handlers
         file_input.change(
@@ -120,7 +125,7 @@ def create_ui():
         
         search_btn.click(
             fn=app.search_documents,
-            inputs=[state, query_input, num_results],
+            inputs=[state, query_input],
             outputs=[images, llm_answer]
         )