Update app.py

app.py

@@ -188,25 +188,21 @@ def generate_download_button(file, label, mime_type):
     """
 
 # --- Streamlit UI ---
-st.set_page_config("Deep Research Bot", layout="centered")
+st.set_page_config("Deep Research Assistant", layout="centered")
 
-# Inject Mermaid.js for mind map rendering
+# Mermaid.js for mind map rendering
 st.markdown("""
-    <script src="https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.min.js"></script>
-    <script>
-        mermaid.initialize({ startOnLoad: true });
-    </script>
+<script src="https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.min.js"></script>
+<script>
+    mermaid.initialize({ startOnLoad: true });
+</script>
 """, unsafe_allow_html=True)
 
 st.markdown("""
-    <style>
+<style>
     .stApp { background-color: #0f172a; color: white; }
     h1, h2, h3 { color: #facc15; }
-    .css-1d391kg, .css-1kyxreq, .css-q8sbsg {
-        background-color: #1e293b; color: white;
-        border-radius: 10px; padding: 10px;
-    }
-    </style>
+</style>
 """, unsafe_allow_html=True)
 
 with st.sidebar:
@@ -215,8 +211,8 @@ with st.sidebar:
     report_type = st.selectbox("📄 Report Type", ["Summary", "Detailed Report", "Thorough Academic Research"])
     tone = st.selectbox("🎯 Tone", ["Objective", "Persuasive", "Narrative"])
     source_type = st.selectbox("📚 Sources", ["Web Only", "Academic Only", "Hybrid"])
-    custom_domains = st.text_input("🌐 Optional Web Domains", placeholder="example.com, techcrunch.com")
-    research_button = st.button("🔎 Run Deep Research")
+    custom_domains = st.text_input("🌐 Optional Web Domains", placeholder="example.com, forbes.com")
+    research_button = st.button("🚀 Run Deep Research")
 
 st.title("📘 Research Output")
 
@@ -228,9 +224,8 @@ if research_button and topic:
                 if source_type in ["Web Only", "Hybrid"]:
                     sources += get_sources(topic, custom_domains)
                 if source_type in ["Academic Only", "Hybrid"]:
-                    status.update(label="📚 Fetching academic sources...")
-                    sources += get_arxiv_papers(topic) or []
-                    sources += get_semantic_papers(topic) or []
+                    sources += get_arxiv_papers(topic)
+                    sources += get_semantic_papers(topic)
                 return sources
 
             all_sources, retries = [], 0
@@ -239,16 +234,14 @@ if research_button and topic:
                 if all_sources:
                     break
                 retries += 1
-                status.update(label=f"🔁 Retrying... ({retries}) Deeper research underway...")
                 time.sleep(2)
 
             if not all_sources:
-                raise ValueError("Unable to fetch any sources. Please try again later.")
+                raise ValueError("❌ No sources found.")
 
             merged = merge_duplicates(all_sources)
 
-        # Image previews
-        st.markdown("---")
+        # 🔹 Image previews
         st.subheader("🖼 Source Previews")
         image_shown = False
         cols = st.columns(2)
@@ -258,29 +251,20 @@ if research_button and topic:
                     st.image(m["image_url"], caption=m["title"], use_container_width=True)
                     image_shown = True
         if not image_shown:
-            st.info("ℹ️ No image previews available from sources.")
+            st.info("ℹ️ No image previews available.")
 
+        # 🔹 Generate report
         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]
-        )
-
-        st.status("🧠 Synthesizing final report...")
+        combined_text = "\n\n".join([
+            f"- [{m['title']}]({m['url']})\n> {m.get('snippet', m.get('summary', ''))[:300]}..."
+            for m in merged
+        ])
 
         prompt = f"""
 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
-- Identified Research Gap
-- Proposed Novel Topic
-- Methodology
-- Comparative Insight
-- Future Work
-- Citations (APA)
+1. Analyze the following sources.
+2. Identify research gaps and propose a novel topic.
+3. Write a {report_type.lower()} in a {tone.lower()} tone.
 
 Sources:
 {combined_text}
@@ -289,7 +273,6 @@ APA Citations:
 {chr(10).join(citations)}
         """
 
-        # Generate report
         st.subheader(f"📝 {report_type} on '{topic}'")
         output_placeholder = st.empty()
         full_output = ""
@@ -297,68 +280,57 @@ APA Citations:
             full_output += chunk
             output_placeholder.markdown(full_output, unsafe_allow_html=True)
 
-        # Store report for mind map and follow-up
+        # Save output for follow-up
         st.session_state["last_report"] = full_output
 
-        # Downloads
-        st.markdown("---")
+        # 🔹 Downloads
         st.subheader("📄 Downloads")
         st.markdown(generate_download_button(generate_pdf(full_output), "Research_Report.pdf", "application/pdf"), unsafe_allow_html=True)
         st.markdown(generate_download_button(generate_latex(full_output), "Research_Report.tex", "application/x-latex"), unsafe_allow_html=True)
 
-        # Plagiarism
-        st.markdown("---")
+        # 🔹 Plagiarism Check
         st.subheader("🔎 Plagiarism Check")
         overlaps = check_plagiarism(full_output, topic)
         if overlaps:
-            st.warning("⚠️ Overlaps found:")
+            st.warning("⚠️ Overlap Detected:")
             for hit in overlaps:
                 st.markdown(f"- [{hit['title']}]({hit['url']})")
         else:
-            st.success("✅ No significant overlaps found.")
+            st.success("✅ No major overlaps found.")
 
-        # Mind Map Generator (Visual)
-        st.markdown("---")
+        # 🔹 Mind Map Generator
         st.subheader("🧭 Visual Mind Map")
-        
-        if "last_report" not in st.session_state or not st.session_state["last_report"].strip():
-            st.warning("⚠️ Please generate a report first.")
-        else:
-            if st.button("🗺 Generate Mind Map"):
+        if st.button("🗺 Generate Mind Map", key="mindmap_btn"):
+            if "last_report" in st.session_state:
                 mindmap_prompt = [
-                    {"role": "system", "content": "You are a researcher who converts reports into mermaid mind maps."},
-                    {"role": "user", "content": f"Create a mermaid.js mind map of the following:\n\n{st.session_state['last_report']}"}
+                    {"role": "system", "content": "Convert this research report into a mermaid.js mind map."},
+                    {"role": "user", "content": st.session_state["last_report"]}
                 ]
                 mindmap_code = ""
                 for chunk in call_llm(mindmap_prompt):
                     mindmap_code += chunk
-        
-                # Remove wrapping markdown if exists
                 mindmap_code = mindmap_code.replace("```mermaid", "").replace("```", "").strip()
-        
-                # Render mermaid directly
                 st.markdown(f"<div class='mermaid'>{mindmap_code}</div>", unsafe_allow_html=True)
+            else:
+                st.error("❌ No report found. Please run research first.")
 
-
-        # Follow-up Q&A
-        st.markdown("---")
-        st.subheader("💬 Ask a Follow-Up Question")
-        follow_input = st.text_input("Ask a question related to the report above:")
+        # 🔹 Follow-up Interaction
+        st.subheader("💬 Ask a Follow-Up")
+        follow_input = st.text_input("Ask a question about the report:")
         if st.button("🔄 Submit Follow-Up"):
             if "last_report" in st.session_state:
-                with st.spinner("Thinking..."):
-                    follow_prompt = [
-                        {"role": "system", "content": "You are a helpful research assistant."},
-                        {"role": "user", "content": f"Here is the report:\n{st.session_state['last_report']}"},
-                        {"role": "user", "content": follow_input}
-                    ]
-                    follow_output = ""
-                    follow_placeholder = st.empty()
-                    for chunk in call_llm(follow_prompt):
-                        follow_output += chunk
-                        follow_placeholder.markdown(follow_output, unsafe_allow_html=True)
+                follow_prompt = [
+                    {"role": "system", "content": "You are a helpful academic assistant."},
+                    {"role": "user", "content": st.session_state["last_report"]},
+                    {"role": "user", "content": follow_input}
+                ]
+                follow_output = ""
+                follow_box = st.empty()
+                for chunk in call_llm(follow_prompt):
+                    follow_output += chunk
+                    follow_box.markdown(follow_output, unsafe_allow_html=True)
             else:
-                st.error("⚠️ Report not found. Please generate a report first.")
+                st.warning("⚠️ Please generate the report first.")
 
     except Exception as e:
-        st.error(f"❌ Error occurred: {e}")
+        st.error(f"❌ Error occurred: {e}")