Update app.py

app.py

@@ -114,15 +114,30 @@ Answer:"""
             # Perform search
             search_results = self.search(question)
             
+            if self.debug:
+                print(f"Search results type: {type(search_results)}")
+                print(f"Search results: {search_results}")
+            
             if not search_results:
                 return "No search results found. Let me try to answer based on my knowledge:\n\n" + self._answer_with_llm(question)
 
-            # Format search results
+            # Format search results - handle different result formats
             formatted_results = []
             for i, result in enumerate(search_results[:3]):  # Use top 3 results
-                title = result.get("title", "No title")
-                snippet = result.get("snippet", "").strip()
-                link = result.get("link", "")
+                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 = ""
                 
                 formatted_results.append(f"Result {i+1}:\nTitle: {title}\nContent: {snippet}\nSource: {link}")
 
@@ -151,12 +166,17 @@ Answer:"""
                 
             except Exception as e:
                 # Fallback to simple search result formatting
-                top_result = search_results[0]
-                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 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}"
+                    else:
+                        return f"Search result: {str(top_result)}"
+                else:
+                    return "Search completed but no usable results found."
 
         except Exception as e:
             return f"Search failed: {e}. Let me try to answer based on my knowledge:\n\n" + self._answer_with_llm(question)
@@ -421,6 +441,7 @@ def clear_cache():
     return "Cache cleared successfully.", None
 
 
+
 # --- Enhanced Gradio Interface ---
 with gr.Blocks(title="Intelligent Agent with Conditional Search") as demo:
     gr.Markdown("# Intelligent Agent with Conditional Search")