Update app.py

app.py

@@ -32,21 +32,10 @@ def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3
         "max_tokens": max_tokens,
         "temperature": temperature
     }
-
     response = requests.post(url, headers=headers, json=data)
-
-    try:
-        result = response.json()
-    except Exception:
-        raise RuntimeError(f"LLM API returned invalid JSON: {response.text}")
-
+    result = response.json()
     if response.status_code != 200:
-        error_msg = result.get("error", {}).get("message", "Unknown error from LLM API.")
-        raise RuntimeError(f"LLM API error: {error_msg}")
-
-    if "choices" not in result:
-        raise RuntimeError(f"LLM API returned unexpected response format. Full response:\n{result}")
-
+        raise RuntimeError(result.get("error", {}).get("message", "LLM API error"))
     return result["choices"][0]["message"]["content"]
 
 def get_sources(topic, domains=None):
@@ -54,19 +43,15 @@ def get_sources(topic, domains=None):
     if domains:
         domain_filters = [d.strip() for d in domains.split(",") if d.strip()]
         query += " site:" + " OR site:".join(domain_filters)
-    try:
-        response = tavily.search(query=query, search_depth="advanced", max_results=10)
-        sources = []
-        for item in response.get("results", []):
-            sources.append({
-                "title": item.get("title"),
-                "url": item.get("url"),
-                "snippet": item.get("content", "")
-            })
-        return sources
-    except Exception as e:
-        st.error(f"Error retrieving sources: {e}")
-        return []
+    response = tavily.search(query=query, search_depth="advanced", max_results=10)
+    sources = []
+    for item in response.get("results", []):
+        sources.append({
+            "title": item.get("title"),
+            "url": item.get("url"),
+            "snippet": item.get("content", "")
+        })
+    return sources
 
 def get_arxiv_papers(query):
     from urllib.parse import quote_plus
@@ -136,14 +121,10 @@ def generate_download_button(file, label, mime_type):
     b64 = base64.b64encode(file.read()).decode()
     return f"""
         <a href=\"data:{mime_type};base64,{b64}\" download=\"{label}\">
-            \ud83d\udce5 Download {label}
+            \ud83d\udcbe Download {label}
         </a>
     """
 
-def fetch_related_image(topic):
-    search_url = f"https://source.unsplash.com/800x400/?{topic}"
-    return search_url
-
 # --- Streamlit UI ---
 st.set_page_config("Deep Research Bot", layout="wide")
 
@@ -162,7 +143,7 @@ with st.sidebar:
     ])
     source_type = st.selectbox(" Sources to include", ["Web Only", "Academic Only", "Hybrid"])
     custom_domains = st.text_input(" Query Domains (Optional)", placeholder="techcrunch.com, forbes.com")
-    research_button = st.button(" Start Research")
+    research_button = st.button("Research")
 
 st.title(" Research Output")
 
@@ -194,33 +175,56 @@ if research_button and topic:
                 citations.append(generate_apa_citation(m['title'], m['url'], m['source']))
 
         with st.spinner(" Synthesizing report..."):
-            prompt = f"""
+            if "Summary" in report_type:
+                prompt = f"""
+# Topic Overview: {topic}
+Tone: {tone}
+Sources:
+{combined_text}
+Write a brief summary that introduces the topic, key findings, and general importance. Use markdown.
+                """
+            elif "Detailed Report" in report_type:
+                prompt = f"""
 # Research Topic: {topic}
 Tone: {tone}
-Type: {report_type}
 Sources:
 {combined_text}
-Write the report in academic markdown with paragraphs (use bullet points only when necessary). Include:
+Write a structured report in markdown including:
 1. Introduction
 2. Research Gap
 3. Novel Insight
-4. Application
-5. Full Academic Writeup if Thorough Report
-"""
+4. Path Forward to Bridge the Research Gap
+5. Citations
+                """
+            else:
+                prompt = f"""
+# Thorough Academic Research Paper
+Topic: {topic}
+Tone: {tone}
+Sources:
+{combined_text}
+Write a detailed research paper in academic markdown with these sections:
+1. Abstract
+2. Introduction
+3. Literature Review
+4. Research Gap
+5. Proposed Methodology or Novel Insight
+6. Applications and Implications
+7. Conclusion
+8. References in APA format
+Also, suggest 1-2 relevant open-license images and include their links.
+                """
+
             final_output = call_llm([{"role": "user", "content": prompt}])
 
         st.markdown(f" {report_type}")
         st.markdown(final_output, unsafe_allow_html=True)
 
-        if report_type == "Thorough Academic Research (~10 min)":
-            image_url = fetch_related_image(topic)
-            st.image(image_url, caption="Related Image", use_column_width=True)
-
-        st.markdown(" Citations (APA Format)")
+        st.markdown("Citations (APA Format)")
         for cite in citations:
             st.markdown(f"- {cite}")
 
-        if report_type == "Thorough Academic Research (~10 min)":
+        if "Thorough Academic Research" in report_type:
             with st.spinner(" Preparing PDF and LaTeX..."):
                 pdf_file = generate_pdf(final_output)
                 latex_file = generate_latex(final_output)
@@ -236,4 +240,4 @@ Write the report in academic markdown with paragraphs (use bullet points only wh
             st.success(" No major overlaps found.")
 
     except Exception as e:
-        st.error(f"Error: {e}")
+        st.error(f"Error: {e}")