Update app.py

app.py

@@ -3,9 +3,8 @@ import gradio as gr
 from azure.storage.blob import BlobServiceClient
 from azure.search.documents import SearchClient
 from azure.search.documents.models import QueryType
-from azure.core.credentials import AzureKeyCredential  # Correct credential import
-from azure.ai.openai import OpenAIClient
-from azure.identity import DefaultAzureCredential
+from azure.core.credentials import AzureKeyCredential
+import requests  # Added for OpenAI API call
 
 # Azure Configuration
 SEARCH_ENDPOINT = "https://rag-search-service-ak.search.windows.net"
@@ -18,7 +17,22 @@ BLOB_CONTAINER_NAME = "dataragak"
 
 # Azure Clients
 search_client = SearchClient(endpoint=SEARCH_ENDPOINT, index_name=INDEX_NAME, credential=AzureKeyCredential(SEARCH_KEY))
-openai_client = OpenAIClient(credential=DefaultAzureCredential(), endpoint=OPENAI_ENDPOINT)
+
+# Function: Call OpenAI API using Requests
+def call_openai_api(prompt):
+    url = f"{OPENAI_ENDPOINT}openai/deployments/gpt-4/completions?api-version=2023-05-15"
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {OPENAI_KEY}"
+    }
+    data = {
+        "prompt": prompt,
+        "max_tokens": 500,
+        "temperature": 0.7
+    }
+    response = requests.post(url, headers=headers, json=data)
+    response_json = response.json()
+    return response_json.get("choices", [{}])[0].get("text", "No response from OpenAI API.")
 
 # Function: Process User Query
 def process_query(question, history):
@@ -29,14 +43,9 @@ def process_query(question, history):
     )
     content = "\n".join([result['content'] for result in search_results])
 
-    openai_response = openai_client.completions.create(
-        deployment_id="gpt-4",
-        prompt=f"Use the following content to answer the question:\n{content}\nQuestion: {question}",
-        max_tokens=500,
-        temperature=0.7
-    )
-
-    return openai_response.choices[0].text.strip()
+    # Call OpenAI API
+    prompt = f"Use the following content to answer the question:\n{content}\nQuestion: {question}"
+    return call_openai_api(prompt)
 
 # Function: Upload PDFs and Reindex
 def upload_pdfs(file_list):
@@ -48,9 +57,6 @@ def upload_pdfs(file_list):
         blob_client = container_client.get_blob_client(blob_name)
         blob_client.upload_blob(file, overwrite=True)
 
-    # Trigger reindexing (update indexer in Azure Cognitive Search)
-    # Note: Use Azure SDK or REST API to trigger the indexer.
-
     return "PDFs uploaded successfully! Reindexing may take some time."
 
 # Gradio UI