Update app.py

app.py

@@ -4,7 +4,11 @@ import time
 from cerebras.cloud.sdk import Cerebras
 
 # Set up the Cerebras client
-client = Cerebras(api_key=os.getenv("CEREBRAS_API_KEY"))
+api_key = os.getenv("CEREBRAS_API_KEY")
+if not api_key:
+    raise ValueError("CEREBRAS_API_KEY environment variable is not set.")
+
+client = Cerebras(api_key=api_key)
 
 def chat_with_cerebras(user_input):
     """
@@ -32,10 +36,11 @@ def chat_with_cerebras(user_input):
         response = ""
         chain_of_thought = ""
         for chunk in stream:
-            if chunk.choices[0].delta.content:
-                response += chunk.choices[0].delta.content
-            if "Chain of Thought:" in chunk.choices[0].delta.content:
-                chain_of_thought += chunk.choices[0].delta.content.split("Chain of Thought:", 1)[-1]
+            if chunk.choices[0].delta and chunk.choices[0].delta.content:
+                content = chunk.choices[0].delta.content
+                response += content
+                if "Chain of Thought:" in content:
+                    chain_of_thought += content.split("Chain of Thought:", 1)[-1]
 
         # End compute time measurement
         compute_time = time.time() - start_time
@@ -58,10 +63,14 @@ def gradio_ui():
                 chain_of_thought_display = gr.Textbox(label="Chain of Thought", interactive=False, lines=10)
 
         user_input = gr.Textbox(label="Type your message", placeholder="Ask me anything...", lines=2)
-        send_button = gr.Button("Send", variant="primary")
-        clear_button = gr.Button("Clear Chat")
+
+        with gr.Row():
+            send_button = gr.Button("Send", variant="primary")
+            clear_button = gr.Button("Clear Chat")
 
         def handle_chat(chat_history, user_input):
+            if not user_input.strip():
+                return chat_history, "", "Please enter a valid message."
             ai_response, chain_of_thought, compute_info = chat_with_cerebras(user_input)
             chat_history.append((user_input, ai_response))
             return chat_history, chain_of_thought, compute_info
@@ -72,6 +81,8 @@ def gradio_ui():
         send_button.click(handle_chat, inputs=[chat_history, user_input], outputs=[chat_history, chain_of_thought_display, compute_time])
         clear_button.click(clear_chat, outputs=[chat_history, chain_of_thought_display, compute_time])
 
+        user_input.submit(handle_chat, inputs=[chat_history, user_input], outputs=[chat_history, chain_of_thought_display, compute_time])
+
         gr.Markdown("""---\n### 🌟 Features:\n- **Advanced Reasoning**: Chain-of-thought explanations for complex queries.\n- **Real-Time Performance Metrics**: Measure response compute time instantly.\n- **Insightful Chain of Thought**: See the reasoning process behind AI decisions.\n- **User-Friendly Design**: Intuitive chatbot interface with powerful features.\n- **Powered by IntellijMind**: Setting new standards for AI interaction.\n""")
 
     return demo