Update app.py

app.py

@@ -18,12 +18,17 @@ logger = logging.getLogger(__name__)
 def setup_openai_auth():
     api_key = os.environ.get("LLM_API_KEY")
     if not api_key:
-        raise Exception("LLM API authentication failed. Please set your API key.")
+        raise Exception("LLM API authentication failed. Please set your LLM_API_KEY environment variable.")
     client = OpenAI(api_key=api_key)
     return client
 
 # Initialize OpenAI client
-client = setup_openai_auth()
+try:
+    client = setup_openai_auth()
+    logger.info("OpenAI client initialized successfully.")
+except Exception as e:
+    logger.error(f"Failed to initialize OpenAI client: {e}")
+    client = None  # Handle gracefully in the application
 
 class OpenAILLMHandler:
     def __init__(self):
@@ -40,6 +45,10 @@ class OpenAILLMHandler:
             logger.info("Returning cached response.")
             return cached_response
 
+        if not self.client:
+            logger.error("OpenAI client is not initialized.")
+            return "Error: AI service is unavailable."
+
         try:
             with self.generation_lock:
                 response = self.client.chat.completions.create(
@@ -99,9 +108,9 @@ def generate_response(user_message: str, conversation_history: list) -> str:
     
     prompt_parts = [generate_system_message(current_time, now)]
     
-    for message in conversation_history:
-        prefix = "User: " if message["role"] == "user" else "Assistant: "
-        prompt_parts.append(f"{prefix}{message['content']}")
+    for user_msg, bot_msg in conversation_history:
+        prompt_parts.append(f"User: {user_msg}")
+        prompt_parts.append(f"Assistant: {bot_msg}")
     
     prompt_parts.append(f"User: {user_message}\nAssistant:")
     prompt = "\n\n".join(prompt_parts)
@@ -113,12 +122,11 @@ def chatbot_interface(user_message: str, history: list) -> list:
         return history
     
     if not llm_handler.model:
-        history.append({"role": "system", "content": "Error: Please call the Human first."})
+        history.append(("System", "Error: AI service is unavailable."))
         return history
     
     ai_response = generate_response(user_message, history)
-    history.append({"role": "user", "content": user_message})
-    history.append({"role": "assistant", "content": ai_response})
+    history.append((user_message, ai_response))
     return history
 
 # Enhanced Gradio UI with improved CSS and layout
@@ -151,13 +159,11 @@ body, .gradio-container {
 .user-message {
     background-color: #d1e7dd;
     align-self: flex-end;
-    margin-left: auto;
 }
 
 .bot-message {
     background-color: #f8d7da;
     align-self: flex-start;
-    margin-right: auto;
 }
 
 #textbox {
@@ -167,7 +173,7 @@ body, .gradio-container {
     border-radius: 5px;
 }
 
-#send-button, #load-button {
+#send-button {
     background-color: #0d6efd;
     color: white;
     border: none;
@@ -177,7 +183,7 @@ body, .gradio-container {
     margin-left: 10px;
 }
 
-#send-button:hover, #load-button:hover {
+#send-button:hover {
     background-color: #0b5ed7;
 }
 
@@ -185,16 +191,20 @@ body, .gradio-container {
     background-color: #6c757d !important;
     cursor: not-allowed;
 }
+
+#model-status {
+    display: none;  /* Hide the model status as "Call Human" is removed */
+}
 """
 
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("<h1 style='text-align: center; color: #0d6efd;'>Human.</h1>")
     
     with gr.Row():
-        load_button = gr.Button("Call Human", variant="primary", elem_id="load-button")
+        # Removed the "Call Human" button
         model_status = gr.Textbox(
             label="Human Arrival Status",
-            value="Human Not Listening.",
+            value="",  # Empty since the button is removed
             interactive=False,
             elem_id="model-status"
         )
@@ -216,35 +226,17 @@ with gr.Blocks(css=custom_css) as demo:
                 )
                 send = gr.Button("➤", elem_id="send-button")
     
-    def load_model_click():
-        if llm_handler.model:
-            return "Human Already Listening."
-        try:
-            # Reload the model name from environment if needed
-            llm_handler.model = os.environ.get("MODEL_NAME", "gpt-4")
-            if not llm_handler.model:
-                return "Failed to load model. Please set the MODEL_NAME environment variable."
-            return "Human Called Successfully."
-        except Exception as e:
-            logger.error(f"Error loading model: {e}")
-            return f"Error loading model: {str(e)}"
-    
     def update_chat(user_message, history):
         if not user_message.strip():
             return history, gr.update(value="")
         if not llm_handler.model:
-            history.append({"role": "system", "content": "Error: Please call the Human first."})
+            history.append(("System", "Error: AI service is unavailable."))
             return history, gr.update(value="")
             
         updated_history = chatbot_interface(user_message, history)
         return updated_history, gr.update(value="")
     
     # Event handlers
-    load_button.click(
-        load_model_click,
-        outputs=[model_status]
-    )
-
     send.click(
         update_chat,
         inputs=[msg, chatbot],
@@ -258,4 +250,7 @@ with gr.Blocks(css=custom_css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    if client:
+        demo.launch(share=True)
+    else:
+        logger.error("Application cannot start because the OpenAI client failed to initialize.")