Update app.py

app.py

@@ -325,19 +325,17 @@ def ask_question(history, question):
     if "similar analysis" in question.lower() and (last_uploaded_company or "").lower() != "apple":
         similar_prompts = [item for item in prompt_history if item["company"].lower() == "apple"]
         if not similar_prompts:
-            return history + [(question, "No Apple prompts stored to reference.")], gr.update()
+            return history + [(question, "No Apple prompts stored to reference.")], gr.update(value=None)
     
         query_embedding = embedding_model.encode(question)
         scores = [np.dot(query_embedding, item["embedding"]) for item in similar_prompts]
         top_idxs = np.argsort(scores)[::-1][:3]
     
-        # optional: rewrite "Apple" -> current company just for nicer text
         def rewrite_company(text: str) -> str:
             company = (last_uploaded_company or "this company").title()
             return text.replace("Apple's", f"{company}'s").replace("Apple", company)
     
-        # quick support checks to skip prompts that the current file can't satisfy
-        df = global_df  # use the currently uploaded file
+        df = global_df
         def dataset_supports(prompt: str) -> bool:
             q = prompt.lower()
             if "research and development" in q or "r&d" in q:
@@ -346,9 +344,10 @@ def ask_question(history, question):
                     or df["tag"].astype(str).str.contains("Research.*Development|TechnologyAndContent", case=False, na=False).any()
                 )
                 return bool(has_rnd)
-            # add similar guards for other analyses if you like
             return True
     
+        image_to_show = None  # <— NEW
+    
         for i in top_idxs:
             past_q = similar_prompts[i]["question"]
             if not dataset_supports(past_q):
@@ -357,11 +356,17 @@ def ask_question(history, question):
             try:
                 resp, img = handle_custom_question(rewrite_company(past_q))
                 history.append((f"Reused: {rewrite_company(past_q)}", resp))
+                if img is not None:
+                    image_to_show = img  # <— capture latest non-None image
             except Exception as e:
                 history.append((f"Skipped (error): {past_q}", f"Couldn’t run this on {last_uploaded_company.title()}: {e}"))
                 continue
     
-        return history, gr.update()
+        # Return the captured image if any; otherwise clear the image pane
+        if image_to_show is not None:
+            return history, image_to_show
+        else:
+            return history, gr.update(value=None)
 
     # === Handle current question ===
     custom_response, custom_image = handle_custom_question(question)