Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import time
 from cerebras.cloud.sdk import Cerebras
+import markdown
 
 # Set up the Cerebras client
 client = Cerebras(api_key=os.getenv("CEREBRAS_API_KEY"))
@@ -37,19 +38,25 @@ 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
 
-        return response, chain_of_thought, f"Compute Time: {compute_time:.2f} seconds"
+        # Improved formatting for chain of thought
+        formatted_response = response
+        if chain_of_thought:
+           formatted_response += f"\n\n**Chain of Thought:**\n{chain_of_thought}"
 
-    except Exception as e:
-       return f"Error: {str(e)}", "", "An error occurred. Please check your API key or the Cerebras service."
 
+        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."
 
 # Gradio interface
 def gradio_ui():
     with gr.Blocks() as demo:
-        gr.Markdown("""# 🚀 IntellijMind: The Future of AI Chatbots\nExperience the most advanced chatbot for deep insights and unmatched clarity!""")
+        gr.Markdown("""# 🚀 IntellijMind Release 1st \nExperience the most advanced chatbot for deep insights and unmatched clarity!""")
 
         with gr.Row():
             with gr.Column(scale=6):
@@ -58,26 +65,26 @@ def gradio_ui():
                 compute_time = gr.Textbox(label="Compute Time", interactive=False)
                 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")
 
         # 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."
+        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."
         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))  # Add user message immediately
-           yield chat_history, "", "Thinking..."
+            chat_history.append((user_input, None))
+            yield chat_history, "", "Thinking..."
 
-           ai_response, chain_of_thought, compute_info = chat_with_cerebras(user_input, default_system_prompt, default_model, default_temperature, default_top_p, default_max_tokens)
-           chat_history[-1] = (user_input, ai_response) # Update the chatbot output with AI response
-           yield chat_history, chain_of_thought, compute_info
+            ai_response, chain_of_thought, compute_info = chat_with_cerebras(user_input, default_system_prompt, default_model, default_temperature, default_top_p, default_max_tokens)
 
+            chat_history[-1] = (user_input, markdown.markdown(ai_response))  # render markdown output to HTML
+            yield chat_history, chain_of_thought, compute_info
 
         def clear_chat():
             return [], "", ""
@@ -88,13 +95,14 @@ def gradio_ui():
             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])
 
+        clear_button.click(clear_chat, 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""")
+        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 Release 1st**: Setting new standards for AI interaction.\n""")
 
     return demo
 
+
 # Run the Gradio app
 demo = gradio_ui()
 demo.launch()