Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
-import openai
+from openai import OpenAI
 import os
 import json
 
-# Set OpenAI API key and base URL from environment variables
-openai.api_key = os.environ["OPENAI_API_KEY"]
-openai.base_url = os.environ["OPENAI_BASE_URL"]
+# Initialize OpenAI client with API key and base URL from environment variables
+client = OpenAI(
+    api_key=os.environ["OPENAI_API_KEY"],
+    base_url=os.environ["OPENAI_BASE_URL"]
+)
 
 # Define the number of results per page and total results to generate
 RESULTS_PER_PAGE = 10
@@ -24,13 +26,13 @@ def fetch_search_results(query):
     """
     
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gemini-2.0-flash-lite",  # Adjust model name as needed
             messages=[
                 {"role": "system", "content": "You are a helpful search engine."},
                 {"role": "user", "content": prompt}
             ],
-            response_format="json_object"
+            response_format={"type": "json_object"}  # Updated for latest SDK
         )
         
         content = response.choices[0].message.content
@@ -46,12 +48,8 @@ def fetch_search_results(query):
         
         return results, None
     
-    except openai.error.OpenAIError as e:
-        return None, f"Error: {str(e)}"
-    except json.JSONDecodeError:
-        return None, "Error: Failed to parse JSON response."
     except Exception as e:
-        return None, f"Unexpected error: {str(e)}"
+        return None, f"Error: {str(e)}"
 
 def display_search_results(query, page=1):
     """Display search results for the given query and page number."""
@@ -108,8 +106,6 @@ def display_search_results(query, page=1):
         html += f'<button onclick="update_page({page + 1})">Next</button>'
     html += '</div></div>'
     
-    # Note: Gradio doesn't support interactive JS directly in HTML outputs,
-    # so we return page numbers for button functionality
     return html, page - 1 if page > 1 else None, page + 1 if page < total_pages else None
 
 def search_handler(query, page):
@@ -139,7 +135,7 @@ with gr.Blocks(title="LLM Search Engine") as app:
         outputs=[output_html, page_state]
     )
     
-    # Note: For full pagination, we simulate Previous/Next with additional buttons
+    # Pagination buttons
     with gr.Row():
         prev_button = gr.Button("Previous", visible=False)
         next_button = gr.Button("Next", visible=False)