Update app.py

app.py

@@ -27,12 +27,12 @@ def fetch_search_results(query):
     
     try:
         response = client.chat.completions.create(
-            model="gemini-2.0-flash-lite",  # Adjust model name as needed
+            model="gemini-2.0-flash-lite",  # Adjust model name as needed (e.g., 'xai-model-name')
             messages=[
                 {"role": "system", "content": "You are a helpful search engine."},
                 {"role": "user", "content": prompt}
             ],
-            response_format={"type": "json_object"}  # Updated for latest SDK
+            response_format={"type": "json_object"}
         )
         
         content = response.choices[0].message.content
@@ -49,7 +49,13 @@ def fetch_search_results(query):
         return results, None
     
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        error_msg = str(e)
+        if "404" in error_msg:
+            return None, f"Error 404: Model or endpoint not found. Check OPENAI_BASE_URL ({os.environ['OPENAI_BASE_URL']}) and model name."
+        elif "401" in error_msg:
+            return None, "Error 401: Invalid API key. Check OPENAI_API_KEY."
+        else:
+            return None, f"Error: {error_msg}"
 
 def display_search_results(query, page=1):
     """Display search results for the given query and page number."""
@@ -98,7 +104,7 @@ def display_search_results(query, page=1):
         html += f'<li class="search-result"><h3>{title}</h3><p>{snippet}</p></li>'
     html += "</ul>"
     
-    # Add pagination controls (simulated with buttons)
+    # Add pagination controls
     html += '<div class="pagination">'
     if page > 1:
         html += f'<button onclick="update_page({page - 1})">Previous</button>'
@@ -116,7 +122,7 @@ def search_handler(query, page):
 # Build Gradio interface with Blocks for state management
 with gr.Blocks(title="LLM Search Engine") as app:
     gr.Markdown("# LLM Search Engine")
-    gr.Markdown("Enter a query below to search using a large language model.")
+    gr.Markdown("Enter a query below to search using a large language model (press Enter or click Search).")
     
     query_input = gr.Textbox(label="Search Query", placeholder="Type your search here...")
     search_button = gr.Button("Search")
@@ -129,6 +135,12 @@ with gr.Blocks(title="LLM Search Engine") as app:
     def on_submit(query, page):
         return search_handler(query, page), page
     
+    # Trigger search on Enter key or button click
+    query_input.submit(
+        fn=on_submit,
+        inputs=[query_input, page_state],
+        outputs=[output_html, page_state]
+    )
     search_button.click(
         fn=on_submit,
         inputs=[query_input, page_state],
@@ -158,8 +170,17 @@ with gr.Blocks(title="LLM Search Engine") as app:
     )
     
     # Update button visibility after search
+    def update_visibility(query, page):
+        html, prev_page, next_page = display_search_results(query, page)
+        return html, page, gr.update(visible=prev_page is not None), gr.update(visible=next_page is not None)
+    
+    query_input.submit(
+        fn=update_visibility,
+        inputs=[query_input, page_state],
+        outputs=[output_html, page_state, prev_button, next_button]
+    )
     search_button.click(
-        fn=lambda q, p: (search_handler(q, p), p, gr.update(visible=p > 1), gr.update(visible=True)),
+        fn=update_visibility,
         inputs=[query_input, page_state],
         outputs=[output_html, page_state, prev_button, next_button]
     )