Update app.py

app.py

@@ -11,7 +11,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 for Ultron AI
+# 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}
@@ -23,10 +23,8 @@ prompt = PromptTemplate(
     template=template
 )
 
-# Memory to keep track of chat history
 memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
-# Language Model configuration
 llm = ChatOpenAI(
     openai_api_key=OPENAI_API_KEY,
     temperature=0.5,
@@ -40,19 +38,50 @@ llm_chain = LLMChain(
     memory=memory
 )
 
-# Response generator function
 def chat_bot_response(user_message, history):
     response = llm_chain.predict(user_message=user_message)
     return response
 
-# Gradio Chat UI
-demo = gr.ChatInterface(
-    fn=chat_bot_response,
-    title="Ultron AI",
-    examples=["How are you doing?", "What are your interests?", "Which places do you like to visit?"],
-    description="Chat with Ultron: a hyper-intelligent, brutally logical assistant for programming, research, cybersecurity, and more."
-)
+with gr.Blocks(css="""
+body {
+    background-color: #0e0e0e;
+    color: #e0e0e0;
+    font-family: 'Orbitron', sans-serif;
+}
+#ultron-head {
+    animation: pulse 2s infinite alternate ease-in-out;
+    margin: auto;
+    display: block;
+    width: 200px;
+    filter: drop-shadow(0 0 10px red);
+}
+@keyframes pulse {
+    0% { transform: rotate(-2deg) scale(1); }
+    100% { transform: rotate(2deg) scale(1.05); }
+}
+.chatbox {
+    background-color: #1a1a1a;
+    border: 2px solid red;
+    padding: 10px;
+    border-radius: 10px;
+}
+""") as demo:
+    gr.HTML("""
+        <h1 style='text-align:center; color:red;'>Ultron AI</h1>
+        <img id="ultron-head" src="https://i.ibb.co/5F3RdjK/ultron-head.png" alt="Ultron Head"/>
+        <p style='text-align:center; color:#aaa;'>Hyper-intelligent AI assistant for logic, code, security & more.</p>
+    """)
+
+    chatbot = gr.Chatbot(label="Ultron", elem_classes="chatbox")
+    msg = gr.Textbox(placeholder="Ask Ultron anything...", label="Your message")
+
+    def respond(user_message, chat_history):
+        response = chat_bot_response(user_message, chat_history)
+        chat_history.append((user_message, response))
+        return "", chat_history
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-# Launch app
+# Launch
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)