Update app.py

app.py

@@ -1,119 +1,183 @@
+import os
 import streamlit as st
-import openai
+from openai import OpenAI
 from typing import List, Dict
+import random
+import base64
 
-# Set up page configuration
+# Advanced Styling and Cosmic Theme
 st.set_page_config(
-    page_title="Grok AI Assistant",
+    page_title="Grok Cosmic Companion",
     page_icon="🚀",
     layout="wide",
     initial_sidebar_state="expanded"
 )
 
-# Custom CSS for enhanced styling
-st.markdown("""
+# Cosmic Background Generator
+def get_cosmic_background():
+    # Create a dynamic, gradient cosmic background
+    colors = [
+        "linear-gradient(135deg, #1e2ad2, #8e2de2)",
+        "linear-gradient(135deg, #ff6a00, #ee0979)",
+        "linear-gradient(135deg, #000428, #004e92)",
+        "linear-gradient(135deg, #2c3e50, #3498db)"
+    ]
+    return random.choice(colors)
+
+# Advanced CSS with Cosmic Design
+st.markdown(f"""
 <style>
-.main-container {
-    background-color: #0E1117;
-    color: #FFFFFF;
-    font-family: 'Roboto', sans-serif;
-}
-.stTextInput>div>div>input {
-    background-color: #262730;
+body {{
+    background: {get_cosmic_background()};
+    color: #ffffff;
+    font-family: 'Orbitron', sans-serif;
+}}
+
+.stApp {{
+    background: transparent;
+}}
+
+.main-container {{
+    background: rgba(0,0,0,0.7);
+    border-radius: 15px;
+    padding: 20px;
+    backdrop-filter: blur(10px);
+}}
+
+.chat-message {{
+    background: rgba(255,255,255,0.1);
+    border-radius: 10px;
+    padding: 15px;
+    margin-bottom: 10px;
+    transition: all 0.3s ease;
+}}
+
+.chat-message:hover {{
+    transform: scale(1.02);
+    box-shadow: 0 0 20px rgba(255,255,255,0.2);
+}}
+
+.stTextInput > div > div > input {{
+    background: rgba(255,255,255,0.1);
     color: white;
-    border: 2px solid #4A4A4A;
-}
-.stButton>button {
-    background-color: #FF4B4B;
+    border: 2px solid rgba(255,255,255,0.2);
+    border-radius: 10px;
+}}
+
+.stButton > button {{
+    background: linear-gradient(45deg, #ff6b6b, #4ecdc4);
     color: white;
     border: none;
     padding: 10px 20px;
-    border-radius: 5px;
+    border-radius: 20px;
     transition: all 0.3s ease;
-}
-.stButton>button:hover {
-    background-color: #FF6B6B;
-    transform: scale(1.05);
-}
-.message-container {
-    background-color: #262730;
-    border-radius: 10px;
-    padding: 15px;
-    margin-bottom: 10px;
-}
-.user-message {
-    background-color: #4A4A4A;
-    text-align: right;
-}
-.ai-message {
-    background-color: #1E1E2E;
-}
+}}
+
+.stButton > button:hover {{
+    transform: scale(1.1);
+    box-shadow: 0 0 20px rgba(255,255,255,0.3);
+}}
 </style>
 """, unsafe_allow_html=True)
 
-# Initialize OpenAI client
-apiKey = st.secrets.get("XAI_API_KEY")
-baseURL: "https://api.x.ai/v1"
-
-class GrokChatApp:
+class AdvancedGrokChatApp:
     def __init__(self):
-        self.messages: List[Dict] = []
-        self.system_prompt = (
-            "You are Grok, a witty AI assistant inspired by the Hitchhiker's Guide to the Galaxy. "
-            "Provide creative, humorous, and insightful responses with a touch of cosmic wisdom."
+        # Cosmic AI Configuration
+        self.XAI_API_KEY = os.getenv("XAI_API_KEY")
+        
+        if not self.XAI_API_KEY:
+            st.error("🛸 Cosmic Connection Lost: API Key Missing!")
+            st.stop()
+        
+        self.client = OpenAI(
+            api_key=self.XAI_API_KEY,
+            base_url="https://api.x.ai/v1"
         )
+        
+        # Advanced Personality Prompts
+        self.personality_modes = {
+            "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."
+        }
+        
+        self.current_mode = "Cosmic Philosopher"
+        self.messages: List[Dict] = []
 
     def generate_response(self, user_input: str) -> str:
         try:
-            # Prepare messages for API call
+            # Enhanced Conversation Context
+            system_prompt = (
+                f"{self.personality_modes[self.current_mode]} "
+                "Respond creatively, with depth and a touch of cosmic wonder."
+            )
+            
             conversation = [
-                {"role": "system", "content": self.system_prompt}
-            ] + [
-                {"role": msg['role'], "content": msg['content']} 
-                for msg in self.messages
-            ] + [
+                {"role": "system", "content": system_prompt}
+            ] + self.messages + [
                 {"role": "user", "content": user_input}
             ]
 
-            # Call Grok AI
-            response = openai.chat.completions.create(
+            response = self.client.chat.completions.create(
                 model="grok-beta",
-                messages=conversation
+                messages=conversation,
+                temperature=0.7,  # More creative responses
+                max_tokens=300
             )
 
             return response.choices[0].message.content
+        
         except Exception as e:
-            return f"🚨 Error: {str(e)}"
+            return f"🌋 Cosmic Disruption: {str(e)}"
 
     def add_message(self, role: str, content: str):
         self.messages.append({"role": role, "content": content})
 
 def main():
-    st.title("🚀 Grok AI: Cosmic Conversations")
+    st.title("🚀 Grok: Cosmic Companion")
     
-    # Initialize chat app
+    # Initialize chat app with session state
     if 'chat_app' not in st.session_state:
-        st.session_state.chat_app = GrokChatApp()
+        st.session_state.chat_app = AdvancedGrokChatApp()
 
-    # Chat history display
-    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'])
+    # Sidebar with Advanced Controls
+    with st.sidebar:
+        st.header("🌌 Cosmic Controls")
+        
+        # Personality Mode Selector
+        mode = st.selectbox(
+            "Choose Grok's Personality", 
+            list(st.session_state.chat_app.personality_modes.keys())
+        )
+        st.session_state.chat_app.current_mode = mode
+        
+        # Conversation Management
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("🔄 Reset"):
+                st.session_state.chat_app.messages = []
+                st.success("Conversation reset to cosmic zero!")
+        
+        with col2:
+            if st.button("💾 Save Conversation"):
+                # Placeholder for conversation saving logic
+                st.info("Conversation saved to cosmic archives!")
 
-    # User input
-    user_input = st.chat_input("Ask Grok anything about life, universe, and everything...")
+    # 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'])
 
-    if user_input:
-        # Display user message
+    # User Input
+    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='👤'):
             st.markdown(user_input)
 
-        # Generate and display AI response
-        with st.chat_message("assistant", avatar='🤖'):
-            with st.spinner("Grok is pondering the mysteries of the universe..."):
+        # AI Response
+        with st.chat_message("assistant", avatar='🚀'):
+            with st.spinner("Traversing cosmic data streams..."):
                 response = st.session_state.chat_app.generate_response(user_input)
                 st.markdown(response)