Update app.py

app.py

@@ -122,26 +122,30 @@ Answer:"""
                 return "No search results found. Let me try to answer based on my knowledge:\n\n" + self._answer_with_llm(question)
 
             # Format search results - handle different result formats
-            formatted_results = []
-            for i, result in enumerate(search_results[:3]):  # Use top 3 results
-                if isinstance(result, dict):
-                    title = result.get("title", "No title")
-                    snippet = result.get("snippet", "").strip()
-                    link = result.get("link", "")
-                elif isinstance(result, str):
-                    # If result is a string, use it directly
-                    title = f"Result {i+1}"
-                    snippet = result
-                    link = ""
-                else:
-                    # Handle other formats
-                    title = f"Result {i+1}"
-                    snippet = str(result)
-                    link = ""
+            if self.debug:
+                print(f"First result type: {type(search_results[0]) if search_results else 'None'}")
+                print(f"First result: {search_results[0] if search_results else 'None'}")
+            
+            # If search_results is a string, use it directly
+            if isinstance(search_results, str):
+                search_context = search_results
+            else:
+                # Handle list of results
+                formatted_results = []
+                for i, result in enumerate(search_results[:3]):  # Use top 3 results
+                    if isinstance(result, dict):
+                        title = result.get("title", "No title")
+                        snippet = result.get("snippet", "").strip()
+                        link = result.get("link", "")
+                        formatted_results.append(f"Title: {title}\nContent: {snippet}\nSource: {link}")
+                    elif isinstance(result, str):
+                        # If result is a string, use it directly
+                        formatted_results.append(result)
+                    else:
+                        # Handle other formats
+                        formatted_results.append(str(result))
                 
-                formatted_results.append(f"Result {i+1}:\nTitle: {title}\nContent: {snippet}\nSource: {link}")
-
-            search_context = "\n\n".join(formatted_results)
+                search_context = "\n\n".join(formatted_results)
 
             # Generate answer using search context
             answer_prompt = f"""You are a helpful AI assistant. Use the provided search results to answer the question accurately. Synthesize information from multiple sources when relevant, and cite sources when appropriate.
@@ -165,16 +169,23 @@ Answer:"""
                 return response.strip()
                 
             except Exception as e:
+                if self.debug:
+                    print(f"LLM generation error: {e}")
                 # Fallback to simple search result formatting
-                if search_results and len(search_results) > 0:
-                    top_result = search_results[0]
-                    if isinstance(top_result, dict):
-                        title = top_result.get("title", "No title")
-                        snippet = top_result.get("snippet", "").strip()
-                        link = top_result.get("link", "")
-                        return f"**{title}**\n\n{snippet}\n\nSource: {link}"
+                if search_results:
+                    if isinstance(search_results, str):
+                        return search_results
+                    elif isinstance(search_results, list) and len(search_results) > 0:
+                        first_result = search_results[0]
+                        if isinstance(first_result, dict):
+                            title = first_result.get("title", "Search Result")
+                            snippet = first_result.get("snippet", "").strip()
+                            link = first_result.get("link", "")
+                            return f"**{title}**\n\n{snippet}\n\n{f'Source: {link}' if link else ''}"
+                        else:
+                            return str(first_result)
                     else:
-                        return f"Search result: {str(top_result)}"
+                        return str(search_results)
                 else:
                     return "Search completed but no usable results found."