Update app.py

app.py

@@ -32,11 +32,21 @@ 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)
-    result = response.json()
+
     if response.status_code != 200:
-        raise RuntimeError(result.get("error", {}).get("message", "LLM API error"))
-    return result["choices"][0]["message"]["content"]
+        try:
+            error_msg = response.json().get("error", {}).get("message", "Unknown error from LLM API.")
+        except Exception:
+            error_msg = response.text
+        raise RuntimeError(f"LLM API error: {error_msg}")
+
+    try:
+        result = response.json()
+        return result["choices"][0]["message"]["content"]
+    except KeyError:
+        raise RuntimeError("LLM API returned unexpected response format. Missing 'choices'.")
 
 def get_sources(topic, domains=None):
     query = topic
@@ -125,6 +135,10 @@ def generate_download_button(file, label, mime_type):
         </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")
 
@@ -143,7 +157,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("Research")
+    research_button = st.button("🚀 Start Research")
 
 st.title("📑 Research Output")
 
@@ -187,12 +201,16 @@ Write the report in academic markdown with paragraphs (use bullet points only wh
 3. Novel Insight
 4. Application
 5. Full Academic Writeup if Thorough Report
-            """
+"""
             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)")
         for cite in citations:
             st.markdown(f"- {cite}")
@@ -213,4 +231,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}")