Spaces:

codelion
/

LLMSearchEngine

Running

App Files Files Community

codelion commited on Feb 28, 2025

Commit

80f4105

verified ·

1 Parent(s): 1cc26a0

Update app.py

Browse files

Files changed (1) hide show

app.py +80 -62

app.py CHANGED Viewed

@@ -2,7 +2,8 @@ import gradio as gr
 from openai import OpenAI
 import os
 import json
-from functools import partial
 # Initialize OpenAI client with API key and base URL from environment variables
 client = OpenAI(
@@ -10,7 +11,7 @@ client = OpenAI(
     base_url=os.environ["OPENAI_BASE_URL"]
 )
-# Define the number of results per page and total results to generate
 RESULTS_PER_PAGE = 10
 TOTAL_RESULTS = 30  # Generate 30 results to allow pagination
@@ -63,6 +64,22 @@ def fetch_search_results(query):
 def generate_search_page(query, page=1):
     """Generate a full HTML search results page."""
     results, error = fetch_search_results(query)
     if error:
@@ -71,6 +88,11 @@ def generate_search_page(query, page=1):
         <head><title>LLM Search Engine</title></head>
         <body style="font-family: Arial, sans-serif;">
             <h1>LLM Search Engine</h1>
             <p style="color: red;">{error}</p>
         </body>
         </html>
@@ -83,11 +105,16 @@ def generate_search_page(query, page=1):
     # Ensure indices are within bounds
     if start_idx >= len(results):
-        return """
         <html>
         <head><title>LLM Search Engine</title></head>
         <body style="font-family: Arial, sans-serif;">
             <h1>LLM Search Engine</h1>
             <p>No more results to display.</p>
         </body>
         </html>
@@ -96,92 +123,85 @@ def generate_search_page(query, page=1):
     paginated_results = results[start_idx:end_idx]
     # Generate full HTML page
-    html = """
     <html>
     <head>
         <title>LLM Search Engine</title>
         <style>
-            body {
                 font-family: Arial, sans-serif;
                 margin: 0;
                 padding: 20px;
                 max-width: 800px;
                 margin-left: auto;
                 margin-right: auto;
-            }
-            .search-box {
-                margin-bottom: 20px;
-            }
-            .search-box input[type="text"] {
                 width: 70%;
                 padding: 8px;
                 font-size: 16px;
                 border: 1px solid #dfe1e5;
                 border-radius: 4px;
-            }
-            .search-box input[type="submit"] {
                 padding: 8px 16px;
                 font-size: 14px;
                 background-color: #f8f9fa;
                 border: 1px solid #dfe1e5;
                 border-radius: 4px;
                 cursor: pointer;
-            }
-            .search-result {
                 margin-bottom: 20px;
-            }
-            .search-result a {
                 color: #1a0dab;
                 font-size: 18px;
                 text-decoration: none;
-            }
-            .search-result a:hover {
                 text-decoration: underline;
-            }
-            .search-result .url {
                 color: #006621;
                 font-size: 14px;
                 margin: 2px 0;
-            }
-            .search-result p {
                 color: #545454;
                 font-size: 14px;
                 margin: 2px 0;
-            }
-            .pagination {
                 margin-top: 20px;
                 text-align: center;
-            }
-            .pagination a {
-                color: #1a0dab;
                 margin: 0 10px;
                 text-decoration: none;
-            }
-            .pagination a:hover {
                 text-decoration: underline;
-            }
-            .pagination span {
-                color: #545454;
-                margin: 0 10px;
-            }
         </style>
     </head>
     <body>
         <h1>LLM Search Engine</h1>
         <form class="search-box" method="get" action="/">
-            <input type="text" name="query" value="{query}" placeholder="Type your search here...">
             <input type="submit" value="Search">
             <input type="hidden" name="page" value="1">
         </form>
         <h2>Results for '{query}' (Page {page} of {total_pages})</h2>
     """
-    # Add search results
     for result in paginated_results:
         title = result.get("title", "No title")
         snippet = result.get("snippet", "No snippet")
         url = result.get("url", "#")
-        html += f"""
         <div class="search-result">
             <a href="{url}" target="_blank">{title}</a>
             <div class="url">{url}</div>
@@ -189,39 +209,37 @@ def generate_search_page(query, page=1):
         </div>
         """
-    # Add pagination
-    html += '<div class="pagination">'
-    if page > 1:
-        html += f'<a href="/?query={query}&page={page - 1}">Previous</a>'
-    else:
-        html += '<span>Previous</span>'
-    html += f'<span>Page {page} of {total_pages}</span>'
-    if page < total_pages:
-        html += f'<a href="/?query={query}&page={page + 1}">Next</a>'
-    else:
-        html += '<span>Next</span>'
-    html += '</div>'
-    html += """
     </body>
     </html>
     """
-    return html
-# Define the app with Blocks
 with gr.Blocks(title="LLM Search Engine") as app:
-    # Custom route handler
-    def handle_request(query, page):
         try:
-            page = int(page) if page else 1
-        except (ValueError, TypeError):
             page = 1
-        return generate_search_page(query, page)
-    # Use a Route to serve raw HTML
-    app.route("/", inputs=[gr.Textbox(visible=False, value=""), gr.Number(visible=False, value=1)],
-               outputs=gr.HTML(), _js="() => [new URLSearchParams(window.location.search).get('query') || '', new URLSearchParams(window.location.search).get('page') || '1']",
-               fn=handle_request)
 app.launch()

 from openai import OpenAI
 import os
 import json
+from urllib.parse import quote
+import html
 # Initialize OpenAI client with API key and base URL from environment variables
 client = OpenAI(
     base_url=os.environ["OPENAI_BASE_URL"]
 )
+# Define constants for pagination
 RESULTS_PER_PAGE = 10
 TOTAL_RESULTS = 30  # Generate 30 results to allow pagination
 def generate_search_page(query, page=1):
     """Generate a full HTML search results page."""
+    if not query.strip():
+        return """
+        <html>
+        <head><title>LLM Search Engine</title></head>
+        <body style="font-family: Arial, sans-serif;">
+            <h1>LLM Search Engine</h1>
+            <form method="get" action="/">
+                <input type="text" name="query" placeholder="Type your search here...">
+                <input type="submit" value="Search">
+                <input type="hidden" name="page" value="1">
+            </form>
+            <p>Please enter a search query.</p>
+        </body>
+        </html>
+        """
     results, error = fetch_search_results(query)
     if error:
         <head><title>LLM Search Engine</title></head>
         <body style="font-family: Arial, sans-serif;">
             <h1>LLM Search Engine</h1>
+            <form method="get" action="/">
+                <input type="text" name="query" value="{html.escape(query)}">
+                <input type="submit" value="Search">
+                <input type="hidden" name="page" value="1">
+            </form>
             <p style="color: red;">{error}</p>
         </body>
         </html>
     # Ensure indices are within bounds
     if start_idx >= len(results):
+        return f"""
         <html>
         <head><title>LLM Search Engine</title></head>
         <body style="font-family: Arial, sans-serif;">
             <h1>LLM Search Engine</h1>
+            <form method="get" action="/">
+                <input type="text" name="query" value="{html.escape(query)}">
+                <input type="submit" value="Search">
+                <input type="hidden" name="page" value="1">
+            </form>
             <p>No more results to display.</p>
         </body>
         </html>
     paginated_results = results[start_idx:end_idx]
     # Generate full HTML page
+    html_content = f"""
     <html>
     <head>
         <title>LLM Search Engine</title>
         <style>
+            body {{
                 font-family: Arial, sans-serif;
                 margin: 0;
                 padding: 20px;
                 max-width: 800px;
                 margin-left: auto;
                 margin-right: auto;
+            }}
+            .search-box input[type="text"] {{
                 width: 70%;
                 padding: 8px;
                 font-size: 16px;
                 border: 1px solid #dfe1e5;
                 border-radius: 4px;
+            }}
+            .search-box input[type="submit"] {{
                 padding: 8px 16px;
                 font-size: 14px;
                 background-color: #f8f9fa;
                 border: 1px solid #dfe1e5;
                 border-radius: 4px;
                 cursor: pointer;
+            }}
+            .search-result {{
                 margin-bottom: 20px;
+            }}
+            .search-result a {{
                 color: #1a0dab;
                 font-size: 18px;
                 text-decoration: none;
+            }}
+            .search-result a:hover {{
                 text-decoration: underline;
+            }}
+            .search-result .url {{
                 color: #006621;
                 font-size: 14px;
                 margin: 2px 0;
+            }}
+            .search-result p {{
                 color: #545454;
                 font-size: 14px;
                 margin: 2px 0;
+            }}
+            .pagination {{
                 margin-top: 20px;
                 text-align: center;
+            }}
+            .pagination a, .pagination span {{
                 margin: 0 10px;
+                color: #1a0dab;
                 text-decoration: none;
+            }}
+            .pagination a:hover {{
                 text-decoration: underline;
+            }}
         </style>
     </head>
     <body>
         <h1>LLM Search Engine</h1>
         <form class="search-box" method="get" action="/">
+            <input type="text" name="query" value="{html.escape(query)}">
             <input type="submit" value="Search">
             <input type="hidden" name="page" value="1">
         </form>
         <h2>Results for '{query}' (Page {page} of {total_pages})</h2>
+        <div class="results">
     """
     for result in paginated_results:
         title = result.get("title", "No title")
         snippet = result.get("snippet", "No snippet")
         url = result.get("url", "#")
+        html_content += f"""
         <div class="search-result">
             <a href="{url}" target="_blank">{title}</a>
             <div class="url">{url}</div>
         </div>
         """
+    # Pagination links
+    encoded_query = quote(query)
+    prev_link = f'<a href="/?query={encoded_query}&page={page-1}">Previous</a>' if page > 1 else '<span>Previous</span>'
+    next_link = f'<a href="/?query={encoded_query}&page={page+1}">Next</a>' if page < total_pages else '<span>Next</span>'
+    html_content += f"""
+        </div>
+        <div class="pagination">
+            {prev_link}
+            <span>Page {page} of {total_pages}</span>
+            {next_link}
+        </div>
     </body>
     </html>
     """
+    return html_content
+# Define the Gradio app with Blocks
 with gr.Blocks(title="LLM Search Engine") as app:
+    def handle_request(request: gr.Request):
+        """Handle the incoming request and return the HTML response."""
+        query = request.query_params.get("query", "")
         try:
+            page = int(request.query_params.get("page", "1"))
+        except ValueError:
             page = 1
+        html = generate_search_page(query, page)
+        return html
+    # Define the route for the root URL
+    app.route("/", fn=handle_request)
+# Launch the app
 app.launch()