Update app.py

app.py

@@ -5,9 +5,8 @@ from langchain.chains import LLMChain
 from langchain_core.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 
-# Get OpenAI API key from environment variable
+# Get OpenAI API key
 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.")
 
@@ -42,38 +41,42 @@ def chat_bot_response(user_message, history):
     response = llm_chain.predict(user_message=user_message)
     return response
 
+# Gradio ChatGPT-style interface using Blocks
 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); }
+    background-color: #f9f9f9;
+    font-family: 'Segoe UI', sans-serif;
 }
 .chatbox {
-    background-color: #1a1a1a;
-    border: 2px solid red;
+    background-color: #ffffff;
+    border: 1px solid #ddd;
+    border-radius: 10px;
     padding: 10px;
+    max-height: 500px;
+    overflow-y: auto;
+}
+.message.user {
+    background-color: #e8f0fe;
+    color: #000;
+    padding: 8px 12px;
+    border-radius: 10px;
+    margin: 5px 0;
+    align-self: flex-end;
+}
+.message.bot {
+    background-color: #f1f1f1;
+    color: #000;
+    padding: 8px 12px;
     border-radius: 10px;
+    margin: 5px 0;
+    align-self: flex-start;
 }
 """) 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>
-    """)
+    
+    gr.HTML("<h1 style='text-align: center;'>🤖 Ultron ChatGPT Style</h1>")
 
-    chatbot = gr.Chatbot(label="Ultron", elem_classes="chatbox")
-    msg = gr.Textbox(placeholder="Ask Ultron anything...", label="Your message")
+    chatbot = gr.Chatbot(elem_classes="chatbox")
+    msg = gr.Textbox(placeholder="Type your message here...", show_label=False)
 
     def respond(user_message, chat_history):
         response = chat_bot_response(user_message, chat_history)
@@ -82,6 +85,6 @@ body {
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-# Launch
+# Run app
 if __name__ == "__main__":
     demo.launch(debug=True)