Update app.py

app.py

@@ -4,8 +4,19 @@ import time
 from cerebras.cloud.sdk import Cerebras
 import markdown
 
-# Set up the Cerebras client
-client = Cerebras(api_key=os.getenv("CEREBRAS_API_KEY"))
+# Function to establish Cerebras client connection
+def get_cerebras_client():
+    try:
+        client = Cerebras(api_key=os.environ.get("CEREBRAS_API_KEY"))
+        return client, None
+    except Exception as e:
+         return None, f"Error connecting to Cerebras: {e}. Please check your API key and ensure network connectivity."
+
+# Cerebras client setup
+client, connection_error = get_cerebras_client()
+if connection_error:
+    print(connection_error)
+    exit() # Exit if the connection failed
 
 def chat_with_cerebras(user_input, system_prompt, model, temperature, top_p, max_completion_tokens):
     """
@@ -38,6 +49,7 @@ def chat_with_cerebras(user_input, system_prompt, model, temperature, top_p, max
             if "Chain of Thought:" in chunk.choices[0].delta.content:
                 chain_of_thought += chunk.choices[0].delta.content.split("Chain of Thought:", 1)[-1]
 
+
         # End compute time measurement
         compute_time = time.time() - start_time
 
@@ -45,11 +57,11 @@ def chat_with_cerebras(user_input, system_prompt, model, temperature, top_p, max
         formatted_response = response
         if chain_of_thought:
            formatted_response += f"\n\n**Chain of Thought:**\n{chain_of_thought}"
-
         return formatted_response, chain_of_thought, f"Compute Time: {compute_time:.2f} seconds"
 
+
     except Exception as e:
-         return f"Error: {str(e)}", "", "An error occurred. Please check your API key or the Cerebras service."
+        return f"Error: {str(e)}", "", "An error occurred while processing your request. Please check the Cerebras service and your input."
 
 # Gradio interface
 def gradio_ui():
@@ -67,14 +79,13 @@ def gradio_ui():
         send_button = gr.Button("Send", variant="primary")
         clear_button = gr.Button("Clear Chat")
 
-         # Set default values for system prompt, model, etc.
+        # Set default values for system prompt, model, etc.
         default_system_prompt = """You are IntellijMind, an advanced AI designed to assist users with detailed insights, problem-solving, and chain-of-thought reasoning. Provide your answers in markdown format. If you do not know the answer, mention that you do not know and don't make things up. Also, remember to be concise and get straight to the point without unnecessary fluff."""
         default_model = "llama-3.3-70b"
         default_temperature = 0.2
         default_top_p = 1
         default_max_tokens = 1024
 
-
         def handle_chat(chat_history, user_input):
             chat_history.append((user_input, None))
             yield chat_history, "", "Thinking..."
@@ -100,6 +111,7 @@ def gradio_ui():
 
     return demo
 
+
 # Run the Gradio app
 demo = gradio_ui()
 demo.launch()