Update app.py

app.py

@@ -29,6 +29,8 @@ def get_cosmic_background() -> str:
 # Advanced CSS with Cosmic Design
 st.markdown(f"""
 <style>
+    @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;500;600;700&display=swap');
+    
     body {{
         background: {get_cosmic_background()};
         color: #ffffff;
@@ -72,16 +74,31 @@ st.markdown(f"""
         transform: scale(1.1);
         box-shadow: 0 0 20px rgba(255,255,255,0.3);
     }}
+    
+    .cosmic-title {{
+        text-align: center;
+        font-size: 2.5rem;
+        margin-bottom: 2rem;
+        text-shadow: 0 0 10px rgba(255,255,255,0.5);
+        animation: glow 2s ease-in-out infinite alternate;
+    }}
+    
+    @keyframes glow {{
+        from {{
+            text-shadow: 0 0 10px rgba(255,255,255,0.5);
+        }}
+        to {{
+            text-shadow: 0 0 20px rgba(255,255,255,0.8),
+                         0 0 30px rgba(255,255,255,0.6);
+        }}
+    }}
 </style>
 """, unsafe_allow_html=True)
 
 class AdvancedGrokChatApp:
     def __init__(self):
-        self.XAI_API_KEY: Optional[str] = os.getenv('XAI_API_KEY')
-        
-        if not self.XAI_API_KEY:
-            st.error("🛸 Cosmic Connection Lost: Please set the XAI_API_KEY environment variable!")
-            st.stop()
+        """Initialize the chat application with API configuration and personality modes."""
+        self.XAI_API_KEY: str = "xai-1HSpHLqxC3LnInrYpwAobgEVsjchUG0PP0adniSXWGQXwq6YfvcPto9MhsS6ouQtC4a4Dh2qqXmERgQQ"
         
         try:
             self.client = OpenAI(
@@ -92,11 +109,11 @@ class AdvancedGrokChatApp:
             st.error(f"🛸 Failed to initialize OpenAI client: {str(e)}")
             st.stop()
         
-        # Advanced Personality Prompts
         self.personality_modes: Dict[str, str] = {
             "Cosmic Philosopher": "You are a wise AI that speaks like a blend of Douglas Adams and Carl Sagan.",
             "Intergalactic Comedian": "You are a witty AI that makes jokes about the universe's absurdities.",
-            "Scientific Oracle": "You provide deep scientific insights with poetic eloquence."
+            "Scientific Oracle": "You provide deep scientific insights with poetic eloquence.",
+            "Space Explorer": "You are an adventurous AI exploring the mysteries of the cosmos."
         }
         
         self.current_mode: str = "Cosmic Philosopher"
@@ -133,13 +150,18 @@ class AdvancedGrokChatApp:
         """Add a message to the conversation history."""
         self.messages.append({"role": role, "content": content})
 
+    def save_conversation(self) -> str:
+        """Save the conversation history to a string format."""
+        return "\n".join([f"{msg['role']}: {msg['content']}" for msg in self.messages])
+
 def initialize_session_state() -> None:
     """Initialize or reset the session state."""
     if 'chat_app' not in st.session_state:
         st.session_state.chat_app = AdvancedGrokChatApp()
 
 def main() -> None:
-    st.title("🚀 Grok: Cosmic Companion")
+    """Main application function."""
+    st.markdown('<h1 class="cosmic-title">🚀 Grok: Cosmic Companion</h1>', unsafe_allow_html=True)
     
     initialize_session_state()
 
@@ -157,26 +179,30 @@ def main() -> None:
         # Conversation Management
         col1, col2 = st.columns(2)
         with col1:
-            if st.button("🔄 Reset"):
+            if st.button("🔄 Reset Chat"):
                 st.session_state.chat_app.messages = []
-                st.success("Conversation reset to cosmic zero!")
+                st.success("✨ Conversation reset to cosmic zero!")
         
         with col2:
             if st.button("💾 Save Chat"):
+                conversation_text = st.session_state.chat_app.save_conversation()
                 st.download_button(
-                    label="Download Chat",
-                    data=str(st.session_state.chat_app.messages),
-                    file_name="cosmic_chat.txt",
+                    label="📥 Download",
+                    data=conversation_text,
+                    file_name="cosmic_conversation.txt",
                     mime="text/plain"
                 )
 
     # Chat Interface
-    for msg in st.session_state.chat_app.messages:
-        with st.chat_message(msg['role'], avatar='🤖' if msg['role'] == 'assistant' else '👤'):
-            st.markdown(msg['content'])
+    chat_container = st.container()
+    
+    with chat_container:
+        for msg in st.session_state.chat_app.messages:
+            with st.chat_message(msg['role'], avatar='🤖' if msg['role'] == 'assistant' else '👤'):
+                st.markdown(msg['content'])
 
     # User Input
-    if user_input := st.chat_input("Whisper your cosmic query..."):
+    if user_input := st.chat_input("🌟 Whisper your cosmic query..."):
         # User Message
         st.session_state.chat_app.add_message("user", user_input)
         with st.chat_message("user", avatar='👤'):
@@ -184,7 +210,7 @@ def main() -> None:
 
         # AI Response
         with st.chat_message("assistant", avatar='🚀'):
-            with st.spinner("Traversing cosmic data streams..."):
+            with st.spinner("🌌 Traversing cosmic data streams..."):
                 response = st.session_state.chat_app.generate_response(user_input)
                 st.markdown(response)