Update app.py

fix retriever bug 1

app.py

@@ -86,6 +86,16 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)", theme=gr.themes.Soft()) a
                     value="Clear All",
                     components=[spec, api_key, out_code]
                 )
+
+    # --- Wrapper function to handle output from the generation agent ---
+    def generate_only(
+        spec, use_rag, top_k, model_choice, api_key, temperature, top_p, max_new_tokens
+    ):
+        # The run_generation function returns three values, but we only want the first one.
+        verilog_code, _, _ = run_generation(
+            spec, use_rag, top_k, model_choice, api_key, temperature, top_p, max_new_tokens
+        )
+        return verilog_code
                 
     def copy_to_clipboard_fn(text):
         return text
@@ -102,7 +112,7 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)", theme=gr.themes.Soft()) a
         outputs=[run_btn, loading_state],
         show_progress=False
     ).then(
-        fn=run_generation,
+        fn=generate_only,
         inputs=[spec, use_rag, top_k, model_choice, api_key, temperature, top_p, max_new_tokens],
         outputs=[out_code],
     ).then(