Spaces:

codelion
/

LLMSearchEngine

Running

App Files Files Community

codelion commited on Feb 28

Commit

3bd40c8

verified ·

1 Parent(s): 009fe9e

Update app.py

Browse files

Files changed (1) hide show

app.py +56 -87

app.py CHANGED Viewed

@@ -69,110 +69,79 @@ def fetch_search_results(query, stream=False):
             return None, f"Error: {error_msg}"
 def stream_search_results(query, page):
-    """Stream search results incrementally."""
     stream_response, error = fetch_search_results(query, stream=True)
     if error:
         yield f"<p style='color: red; text-align: center;'>{error}</p>"
         return
-    # Generate header
-    header = """
     <html>
     <head>
         <title>LLM Search Engine</title>
         <style>
-            body { font-family: Arial, sans-serif; margin: 0; padding: 0; color: #202124; }
-            .header { text-align: center; padding: 20px 0; }
-            .logo { font-size: 36px; font-weight: bold; }
-            .logo span:nth-child(1) { color: #4285f4; }
-            .logo span:nth-child(2) { color: #ea4335; }
-            .logo span:nth-child(3) { color: #fbbc05; }
-            .logo span:nth-child(4) { color: #4285f4; }
-            .logo span:nth-child(5) { color: #34a853; }
-            .search-box { max-width: 584px; margin: 0 auto 20px; }
-            .search-box input[type="text"] {
-                width: 100%; padding: 12px 20px; font-size: 16px;
-                border: 1px solid #dfe1e5; border-radius: 24px;
-                box-shadow: 0 1px 6px rgba(32,33,36,0.28); outline: none;
-            }
-            .search-box input[type="submit"] {
-                background-color: #f8f9fa; border: 1px solid #f8f9fa;
-                border-radius: 4px; color: #3c4043; font-size: 14px;
-                padding: 10px 16px; margin: 11px 4px; cursor: pointer;
-            }
-            .search-box input[type="submit"]:hover {
-                border: 1px solid #dadce0; box-shadow: 0 1px 2px rgba(0,0,0,0.1);
-            }
-            .search-buttons { text-align: center; }
-            .results { max-width: 652px; margin: 0 auto; }
-            .search-result { margin-bottom: 28px; }
-            .search-result a { color: #1a0dab; font-size: 20px; text-decoration: none; }
-            .search-result a:hover { text-decoration: underline; }
-            .search-result .url { color: #006621; font-size: 14px; line-height: 20px; }
-            .search-result p { color: #4d5156; font-size: 14px; line-height: 22px; margin: 0; }
-            .pagination { text-align: center; margin: 40px 0; }
-            .pagination a, .pagination span {
                 color: #1a0dab; font-size: 14px; margin: 0 8px; text-decoration: none;
-            }
-            .pagination a:hover { text-decoration: underline; }
         </style>
     </head>
     <body>
-        <div class="header">
-            <div class="logo">
-                <span>L</span><span>L</span><span>M</span><span> </span><span>Search</span>
-            </div>
-        </div>
-        <div class="search-box">
-            <form method="get" action="/">
-                <input type="text" name="query" value="{{query}}">
-                <input type="hidden" name="page" value="1">
-                <div class="search-buttons">
-                    <input type="submit" name="btn" value="LLM Search">
-                    <input type="submit" name="btn" value="I'm Feeling Lucky">
-                </div>
-            </form>
-        </div>
         <div class="results">
-            <h2 style="font-size: 18px; color: #70757a; margin-bottom: 20px;">Results for '{{query}}' (Page {{page}})</h2>
-    """.replace('{{query}}', html.escape(query)).replace('{{page}}', str(page))
     yield header
-    # Stream and parse results
-    buffer = ""
-    results = []
-    for chunk in stream_response:
-        if chunk.choices[0].delta.content:
-            buffer += chunk.choices[0].delta.content
-            try:
-                # Try to parse the buffer as JSON incrementally
-                temp_results = json.loads(buffer)
-                if isinstance(temp_results, list):
-                    results = temp_results
-                elif isinstance(temp_results, dict) and "results" in temp_results:
-                    results = temp_results["results"]
-                # Process only new results
-                for i, result in enumerate(results[len(results) - (len(results) % RESULTS_PER_PAGE):]):
-                    if i >= start_idx and i < end_idx:
-                        title = html.escape(result.get("title", "No title"))
-                        snippet = html.escape(result.get("snippet", "No snippet"))
-                        url = html.escape(result.get("url", "#"))
-                        yield f"""
-                        <div class="search-result">
-                            <a href="{url}" target="_blank">{title}</a>
-                            <div class="url">{url}</div>
-                            <p>{snippet}</p>
-                        </div>
-                        """
-            except json.JSONDecodeError:
-                continue  # Keep buffering until complete JSON
-    # Calculate pagination
-    start_idx = (page - 1) * RESULTS_PER_PAGE
-    end_idx = start_idx + RESULTS_PER_PAGE
-    total_pages = (len(results) + RESULTS_PER_PAGE - 1) // RESULTS_PER_PAGE
-    # Pagination links
     encoded_query = quote(query)
     prev_link = f'<a href="/?query={encoded_query}&page={page-1}&btn=LLM+Search">Previous</a>' if page > 1 else '<span>Previous</span>'
     next_link = f'<a href="/?query={encoded_query}&page={page+1}&btn=LLM+Search">Next</a>' if page < total_pages else '<span>Next</span>'

             return None, f"Error: {error_msg}"
 def stream_search_results(query, page):
+    """Stream search results incrementally after accumulating the full response."""
     stream_response, error = fetch_search_results(query, stream=True)
     if error:
         yield f"<p style='color: red; text-align: center;'>{error}</p>"
         return
+    # Step 1: Accumulate the full response
+    buffer = ""
+    for chunk in stream_response:
+        if chunk.choices[0].delta.content:
+            buffer += chunk.choices[0].delta.content
+    # Step 2: Parse the full JSON response
+    try:
+        results = json.loads(buffer)
+        if isinstance(results, dict) and "results" in results:
+            results = results["results"]
+        elif not isinstance(results, list):
+            yield "<p style='color: red; text-align: center;'>Error: Unexpected JSON structure.</p>"
+            return
+    except json.JSONDecodeError:
+        yield "<p style='color: red; text-align: center;'>Error: Failed to parse JSON response.</p>"
+        return
+    # Step 3: Calculate pagination
+    RESULTS_PER_PAGE = 10  # Adjust this as needed
+    start_idx = (page - 1) * RESULTS_PER_PAGE
+    end_idx = min(start_idx + RESULTS_PER_PAGE, len(results))
+    total_pages = (len(results) + RESULTS_PER_PAGE - 1) // RESULTS_PER_PAGE
+    # Step 4: Generate and yield the header
+    header = f"""
     <html>
     <head>
         <title>LLM Search Engine</title>
         <style>
+            body {{ font-family: Arial, sans-serif; margin: 0; padding: 0; color: #202124; }}
+            .results {{ max-width: 652px; margin: 0 auto; }}
+            .search-result {{ margin-bottom: 28px; }}
+            .search-result a {{ color: #1a0dab; font-size: 20px; text-decoration: none; }}
+            .search-result a:hover {{ text-decoration: underline; }}
+            .search-result .url {{ color: #006621; font-size: 14px; line-height: 20px; }}
+            .search-result p {{ color: #4d5156; font-size: 14px; line-height: 22px; margin: 0; }}
+            .pagination {{ text-align: center; margin: 40px 0; }}
+            .pagination a, .pagination span {{
                 color: #1a0dab; font-size: 14px; margin: 0 8px; text-decoration: none;
+            }}
+            .pagination a:hover {{ text-decoration: underline; }}
         </style>
     </head>
     <body>
         <div class="results">
+            <h2 style="font-size: 18px; color: #70757a; margin-bottom: 20px;">Results for '{query}' (Page {page} of {total_pages})</h2>
+    """
     yield header
+    # Step 5: Yield each result for the current page
+    for i in range(start_idx, end_idx):
+        if i < len(results):
+            result = results[i]
+            title = result.get("title", "No title")
+            snippet = result.get("snippet", "No snippet")
+            url = result.get("url", "#")
+            yield f"""
+            <div class="search-result">
+                <a href="{url}" target="_blank">{title}</a>
+                <div class="url">{url}</div>
+                <p>{snippet}</p>
+            </div>
+            """
+    # Step 6: Yield pagination links
+    from urllib.parse import quote
     encoded_query = quote(query)
     prev_link = f'<a href="/?query={encoded_query}&page={page-1}&btn=LLM+Search">Previous</a>' if page > 1 else '<span>Previous</span>'
     next_link = f'<a href="/?query={encoded_query}&page={page+1}&btn=LLM+Search">Next</a>' if page < total_pages else '<span>Next</span>'