Update app.py

app.py

@@ -11,6 +11,7 @@ import logging
 from tiktoken import get_encoding
 from openai import AzureOpenAI
 import httpx
+import re
 
 # Configure logging
 logging.basicConfig(
@@ -161,7 +162,14 @@ class TextEditor:
 
 # OpenAIApi class
 class OpenAIApi:
-    def __init__(self, preprompt="", endpoint="https://<your-resource-name>.openai.azure.com/", model="gpt-4o", api_key=None):
+    def __init__(self, preprompt="", endpoint="https://huggingface.co/spaces/Deadmon/copypasta.openai.azure.com/", model="gpt-4o", api_key=None):
+        # Validate endpoint format
+        if "<your-resource-name>" in endpoint:
+            logger.error("Invalid endpoint: Replace '<your-resource-name>' with your Azure OpenAI resource name")
+            raise ValueError("Invalid endpoint: Replace '<your-resource-name>' with your Azure OpenAI resource name (e.g., https://my-resource.openai.azure.com/)")
+        if not re.match(r"^https://[a-zA-Z0-9-]+\.openai\.azure\.com/?$", endpoint):
+            logger.warning(f"Endpoint format may be incorrect: {endpoint}. Expected format: https://<resource-name>.openai.azure.com/")
+        
         # Use a minimal httpx.Client to avoid proxies parameter
         http_client = httpx.Client()
         self.client = AzureOpenAI(
@@ -256,7 +264,7 @@ class OpenAIApi:
         messages.extend({"role": c["role"], "content": c["text"]} for c in context)
         messages.append({"role": "user", "content": sanitized_prompt})
         
-        logger.info(f"Sending request to model: {self.model}, messages: {json.dumps(messages, ensure_ascii=False)}")
+        logger.info(f"Sending request to model: {self.model}, endpoint: {self.client._base_url}, messages: {json.dumps(messages, ensure_ascii=False)}")
         
         try:
             response = await self.client.chat.completions.create(
@@ -309,22 +317,28 @@ class OpenAIApi:
             
         except Exception as e:
             error_msg = f"API Error: {str(e)}"
-            logger.error(f"API request failed: {error_msg}, endpoint: {self.client.azure_endpoint}, model: {self.model}")
+            logger.error(f"API request failed: {error_msg}, endpoint: {self.client._base_url}, model: {self.model}")
             self.memory.add_chunk(error_msg, "system")
             return {"error": error_msg}
 
 # Gradio UI
 async def chat_submit(user_input, chat_history, preprompt):
-    api = OpenAIApi(preprompt=preprompt, api_key=os.getenv("AZURE_OPENAI_API_KEY"))
-    response = await api.fetch_response(user_input)
-    if "error" in response:
-        chat_history.append({"role": "assistant", "content": f"Error: {response['error']}"})
-        logger.warning(f"Chat error: {response['error']}")
-    else:
-        chat_history.append({"role": "user", "content": user_input})
-        chat_history.append({"role": "assistant", "content": response["content"]})
-        logger.info("Chat response added to history")
-    return chat_history, preprompt
+    try:
+        api = OpenAIApi(preprompt=preprompt, api_key=os.getenv("AZURE_OPENAI_API_KEY"))
+        response = await api.fetch_response(user_input)
+        if "error" in response:
+            chat_history.append({"role": "assistant", "content": f"Error: {response['error']}"})
+            logger.warning(f"Chat error: {response['error']}")
+        else:
+            chat_history.append({"role": "user", "content": user_input})
+            chat_history.append({"role": "assistant", "content": response["content"]})
+            logger.info("Chat response added to history")
+        return chat_history, preprompt
+    except ValueError as e:
+        error_msg = f"Configuration Error: {str(e)}"
+        logger.error(error_msg)
+        chat_history.append({"role": "assistant", "content": error_msg})
+        return chat_history, preprompt
 
 def get_history():
     memory = ConversationMemory()
@@ -372,6 +386,7 @@ async def edit_suffix(chunk_id, suffix):
 def create_ui():
     with gr.Blocks(title="Azure OpenAI Chat & Text Editor") as demo:
         gr.Markdown("# Azure OpenAI Chat with Text Editing")
+        gr.Markdown("**Note**: Ensure the Azure OpenAI endpoint is set correctly in `app.py` (replace `<your-resource-name>` with your resource name).")
         
         with gr.Tab("Chat"):
             chatbot = gr.Chatbot(label="Conversation", type="messages")