Update app.py

app.py

@@ -73,12 +73,6 @@ def autonomous_research_agent(topic):
     web = search_duckduckgo(topic)
     images = get_image_urls(topic)
 
-    image_md = ""
-    if images:
-        for img in images:
-            image_md += f"![Image]({img}) "
-        image_md += "\n\n"
-
     arxiv_md = ""
     for p in arxiv:
         arxiv_md += f"- [{p['title']}]({p['url']})\n> {p['summary'][:300]}...\n\n"
@@ -94,8 +88,6 @@ def autonomous_research_agent(topic):
     prompt = f"""
 # Research Topic: {topic}
 
-{image_md}
-
 ## ArXiv:
 {arxiv_md}
 
@@ -121,7 +113,7 @@ Now synthesize this information into:
     if web_md:
         response += "**Web:**\n" + web_md
 
-    return response
+    return response, images
 
 # --- Streamlit UI ---
 st.set_page_config("Autonomous Research Assistant", layout="wide")
@@ -134,10 +126,17 @@ topic = st.text_input("Enter a research topic:")
 if st.button("Run Research Agent"):
     with st.spinner("Gathering sources & thinking..."):
         try:
-            result = autonomous_research_agent(topic)
+            response, images = autonomous_research_agent(topic)
+
+            # Display images
+            if images:
+                st.subheader("🖼️ Relevant Images")
+                st.image(images, width=300)
+
+            # Display markdown response
             st.session_state.chat_history.append({"role": "user", "content": topic})
-            st.session_state.chat_history.append({"role": "assistant", "content": result})
-            st.markdown(result)
+            st.session_state.chat_history.append({"role": "assistant", "content": response})
+            st.markdown(response)
         except Exception as e:
             st.error(f"Failed: {e}")