fixing rag pulling errors

app.py

@@ -222,12 +222,15 @@ def run_generation(spec, use_rag, top_k, model_choice, api_key, temperature, top
     # Retrieval (if enabled)
     docs_with_scores = []
     retrieved_preview = []
+    retrieved_raw_formatted = [] # New list to hold formatted data
     if use_rag:
         try:
             docs_with_scores = agent.vectorstore.similarity_search_with_score(spec, k=top_k)
             for doc, score in docs_with_scores:
                 src = doc.metadata.get("source_file", doc.metadata.get("module", "unknown"))
                 retrieved_preview.append(f"{src}  |  score={score:.4f}")
+                # Add the page content to the new list, formatted as a tuple
+                retrieved_raw_formatted.append((doc.page_content, None))
         except Exception as e:
             return f"// Retrieval error: {e}", "", []
 
@@ -243,8 +246,8 @@ def run_generation(spec, use_rag, top_k, model_choice, api_key, temperature, top
     else:
         code = agent.generate_baseline(spec, gen_params)
 
-    return code.strip(), ("\n".join(retrieved_preview) if retrieved_preview else ""), [d[0].page_content for d in docs_with_scores]
-
+    # Return the new formatted list for the HighlightedText component
+    return code.strip(), ("\n".join(retrieved_preview) if retrieved_preview else ""), retrieved_raw_formatted
 
 with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
     gr.Markdown("## DeepV for RTL Code Generation — Model-Agnostic (Bring Your Own API Key)")