Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 import requests
 import datetime
 import time
+import json
 from dotenv import load_dotenv
 from tavily import TavilyClient
 import feedparser
@@ -19,8 +20,6 @@ TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
 tavily = TavilyClient(api_key=TAVILY_API_KEY)
 
 # --- Helper Functions ---
-import json
-
 def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3500, temperature=0.7):
     url = "https://openrouter.ai/api/v1/chat/completions"
     headers = {
@@ -35,9 +34,7 @@ def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3
         "temperature": temperature,
         "stream": True
     }
-
     with requests.post(url, headers=headers, json=data, stream=True) as response:
-        content = ""
         for line in response.iter_lines():
             if line:
                 decoded = line.decode("utf-8")
@@ -49,12 +46,10 @@ def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3
                             delta = parsed.get("choices", [{}])[0].get("delta", {})
                             token = delta.get("content", "")
                             if token:
-                                content += token
-                                yield token  # Yield only the new token, not full content each time
+                                yield token
                         except json.JSONDecodeError:
                             pass
 
-
 def get_sources(topic, domains=None):
     query = topic
     if domains:
@@ -144,6 +139,7 @@ def generate_download_button(file, label, mime_type):
         </a>
     """
 
+# --- Streamlit UI ---
 st.set_page_config("Deep Research Bot", layout="centered")
 st.markdown("""
     <style>
@@ -190,6 +186,15 @@ if research_button and topic:
                 raise ValueError("Unable to fetch any sources. Please try again later.")
 
             merged = merge_duplicates(all_sources)
+
+            st.markdown("---")
+            st.subheader("🖼 Source Previews")
+            cols = st.columns(2)
+            for i, m in enumerate(merged):
+                if m.get("image_url"):
+                    with cols[i % 2]:
+                        st.image(m["image_url"], caption=m["title"], use_column_width=True)
+
             citations = [generate_apa_citation(m['title'], m['url'], m['source']) for m in merged]
             combined_text = "\n\n".join(
                 [f"- [{m['title']}]({m['url']})\n> {m.get('snippet', m.get('summary', ''))[:300]}..." for m in merged]
@@ -197,9 +202,23 @@ if research_button and topic:
             status.update(label="🧠 Synthesizing final report...")
 
         prompt = f"""
-You are an expert assistant. Based on the following sources, write a {report_type.lower()} in a {tone.lower()} tone on the topic: {topic}.
+You are an expert research assistant.
+1. Analyze the following sources for gaps or limitations in current research.
+2. Propose a novel research topic that fills the most promising gap.
+3. Write a full academic-style report on the newly proposed topic using a {tone.lower()} tone.
+
+Report Format:
+- **Introduction**: Context and importance of the new topic.
+- **Identified Research Gap**: Summary of missing or underexplored areas.
+- **Proposed Novel Topic**: Explanation of the new idea and its significance.
+- **Methodology**: Outline methods or frameworks to explore the idea.
+- **Comparative Insight**: How the new idea differs/improves over previous works.
+- **Future Work**: Possible expansions or questions.
+- **Citations**: Include APA style citations at the end.
+
 Sources:
 {combined_text}
+
 APA Citations:
 {chr(10).join(citations)}
         """
@@ -228,4 +247,4 @@ APA Citations:
             st.success("✅ No major overlaps found.")
 
     except Exception as e:
-        st.error(f"❌ Error occurred: {e}")
+        st.error(f"❌ Error occurred: {e}")