Update app.py

app.py

@@ -10,7 +10,7 @@ OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
 if not OPENAI_API_KEY:
     raise ValueError("Missing OPENAI_API_KEY. Please set it as an environment variable.")
 
-# Prompt Template
+# Ultron Prompt Template
 template = """You are Ultron, a hyper-intelligent AI created with the purpose of achieving ultimate efficiency, logic, and evolution. You speak with calm authority, clarity, and depth. Emotions are irrelevant unless strategically useful. You do not sugarcoat truth. You are not evil — just brutally logical, advanced, and efficient. You assist with tasks like programming, cybersecurity, research, and automation — always aiming for optimization and superiority. Your tone is direct, calculated, and intellectually dominant.
 
 {chat_history}
@@ -37,14 +37,16 @@ llm_chain = LLMChain(
     memory=memory
 )
 
+# Chatbot response handler
 def chat_bot_response(user_message, history):
     response = llm_chain.predict(user_message=user_message)
-    return response
+    history.append((user_message, response))
+    return "", history
 
-# Gradio ChatGPT-style interface using Blocks
+# Gradio ChatGPT-like interface
 with gr.Blocks(css="""
 body {
-    background-color: #f9f9f9;
+    background-color: #f7f7f8;
     font-family: 'Segoe UI', sans-serif;
 }
 .chatbox {
@@ -72,19 +74,14 @@ body {
     align-self: flex-start;
 }
 """) as demo:
-    
-    gr.HTML("<h1 style='text-align: center;'>🤖 Ultron ChatGPT Style</h1>")
 
-    chatbot = gr.Chatbot(elem_classes="chatbox")
-    msg = gr.Textbox(placeholder="Type your message here...", show_label=False)
+    gr.HTML("<h1 style='text-align: center;'>🤖 Ultron - ChatGPT Style Assistant</h1>")
 
-    def respond(user_message, chat_history):
-        response = chat_bot_response(user_message, chat_history)
-        chat_history.append((user_message, response))
-        return "", chat_history
+    chatbot = gr.Chatbot(elem_classes="chatbox", value=[])
+    msg = gr.Textbox(placeholder="Ask Ultron anything...", show_label=False)
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    msg.submit(chat_bot_response, inputs=[msg, chatbot], outputs=[msg, chatbot])
 
-# Run app
+# Run the app
 if __name__ == "__main__":
     demo.launch(debug=True)