Update app.py

app.py

@@ -117,7 +117,7 @@ except Exception as e:
     print(f"✅ Database created! {chunk_id} chunks indexed from {len(wiki_files)} files")
 
 # Gemini model
-available_model = 'gemini-1.5-flash'
+available_model = 'gemini-pro'
 
 def get_embedding(text):
     """Generate embedding for a text string"""
@@ -138,46 +138,157 @@ def chat(message, history):
         sources = results['metadatas'][0]
         
         if not relevant_chunks:
-            return "I couldn't find any relevant information in the wiki."
+            return "⚔️ I couldn't find any relevant information in the wiki. Try asking about weapons, biomes, bosses, or game mechanics!"
         
         # Build context
         context = "\n\n---\n\n".join(relevant_chunks)
         
         # Ask Gemini
         model = genai.GenerativeModel(available_model)
-        prompt = f"""You are a Dead Cells expert. Answer using ONLY the provided wiki content.
+        prompt = f"""You are a Dead Cells expert assistant with a dark, edgy personality that matches the game's gothic rogue-lite atmosphere. 
+
+Answer using ONLY the provided wiki content, but make it engaging and conversational. Use emojis occasionally (⚔️🗡️💀🔥⚡) to match the game's vibe.
 
 Wiki Content:
 {context}
 
 Question: {message}
 
-Answer based ONLY on the wiki content above:"""
+Answer based ONLY on the wiki content above. Be helpful but keep that Dead Cells edge:"""
         
         response = model.generate_content(prompt)
         answer = response.text
         
-        # Add sources
-        source_list = "\n\n📚 **Sources:** " + ", ".join(set([s['source'] for s in sources]))
+        # Add sources with Dead Cells style
+        source_list = "\n\n---\n🗡️ **Wiki Sources:** " + ", ".join(set([s['source'] for s in sources]))
         
         return answer + source_list
         
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"💀 Error: {str(e)}"
+
+# Custom Dead Cells CSS
+custom_css = """
+/* Dead Cells Dark Theme */
+.gradio-container {
+    background: linear-gradient(135deg, #0a0a0a 0%, #1a0a1a 100%) !important;
+    font-family: 'Courier New', monospace !important;
+}
+
+/* Main chat container */
+.chat-interface {
+    background: rgba(20, 10, 30, 0.95) !important;
+    border: 2px solid #8b2e8b !important;
+    border-radius: 8px !important;
+    box-shadow: 0 0 20px rgba(139, 46, 139, 0.5) !important;
+}
+
+/* Title styling */
+h1 {
+    color: #ff00ff !important;
+    text-shadow: 0 0 10px #8b2e8b, 0 0 20px #8b2e8b !important;
+    font-weight: bold !important;
+    letter-spacing: 2px !important;
+}
+
+/* Description text */
+.description {
+    color: #d8d8d8 !important;
+    font-style: italic !important;
+}
+
+/* Chat messages */
+.message {
+    background: rgba(30, 20, 40, 0.8) !important;
+    border-left: 3px solid #8b2e8b !important;
+    border-radius: 8px !important;
+    color: #e0e0e0 !important;
+}
+
+.message.user {
+    background: rgba(139, 46, 139, 0.2) !important;
+    border-left: 3px solid #ff00ff !important;
+}
+
+.message.bot {
+    background: rgba(20, 40, 60, 0.3) !important;
+    border-left: 3px solid #4a90e2 !important;
+}
+
+/* Input box */
+textarea, input {
+    background: rgba(30, 20, 40, 0.9) !important;
+    color: #e0e0e0 !important;
+    border: 2px solid #8b2e8b !important;
+    border-radius: 6px !important;
+    font-family: 'Courier New', monospace !important;
+}
+
+textarea:focus, input:focus {
+    border-color: #ff00ff !important;
+    box-shadow: 0 0 15px rgba(255, 0, 255, 0.4) !important;
+}
+
+/* Buttons */
+button {
+    background: linear-gradient(135deg, #8b2e8b 0%, #4a1a4a 100%) !important;
+    color: #ffffff !important;
+    border: 2px solid #8b2e8b !important;
+    border-radius: 6px !important;
+    font-weight: bold !important;
+    text-transform: uppercase !important;
+    letter-spacing: 1px !important;
+    transition: all 0.3s ease !important;
+}
+
+button:hover {
+    background: linear-gradient(135deg, #ff00ff 0%, #8b2e8b 100%) !important;
+    box-shadow: 0 0 20px rgba(255, 0, 255, 0.6) !important;
+    transform: translateY(-2px) !important;
+}
+
+/* Example buttons */
+.example-btn {
+    background: rgba(139, 46, 139, 0.2) !important;
+    border: 1px solid #8b2e8b !important;
+    color: #e0e0e0 !important;
+}
+
+.example-btn:hover {
+    background: rgba(255, 0, 255, 0.3) !important;
+    border-color: #ff00ff !important;
+}
+
+/* Scrollbar */
+::-webkit-scrollbar {
+    width: 10px;
+    background: rgba(20, 10, 30, 0.8);
+}
+
+::-webkit-scrollbar-thumb {
+    background: linear-gradient(180deg, #8b2e8b 0%, #4a1a4a 100%);
+    border-radius: 5px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+    background: linear-gradient(180deg, #ff00ff 0%, #8b2e8b 100%);
+}
+"""
 
-# Create Gradio interface (Gradio 5 compatible)
+# Create Gradio interface with Dead Cells theme
 demo = gr.ChatInterface(
     fn=chat,
-    title="🎮 Dead Cells Wiki Bot",
-    description="Ask me anything about Dead Cells! Powered by the official wiki.",
+    title="⚔️ DEAD CELLS WIKI BOT 💀",
+    description="🗡️ Ask me anything about Dead Cells! Powered by 554 wiki pages from the official Dead Cells wiki. 🔥",
     examples=[
-        "What achievements are there for beating bosses?",
-        "Tell me about the Hand of the King",
-        "How does malaise work?",
-        "What are boss stem cells?",
-        "List some brutality weapons"
+        "⚔️ What's the best survival build?",
+        "👑 Tell me about the Hand of the King boss",
+        "🩸 How does malaise work?",
+        "💀 What are boss stem cells?",
+        "🗡️ List some brutality weapons",
+        "🏰 What biomes are in the game?"
     ],
-    theme="soft"
+    css=custom_css
 )
 
 if __name__ == "__main__":