Update app.py

app.py

@@ -1,18 +1,59 @@
 import gradio as gr
+import os
+import google.generativeai as genai
 
-# Load your bio and projects
+# Get Gemini API key from Hugging Face Secrets
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+
+# Configure Gemini
+genai.configure(api_key=GEMINI_API_KEY)
+model = genai.GenerativeModel("gemini-1.5-flash")
+
+# Load bio context
 with open("bio.txt", "r") as f:
     bio = f.read()
 
-def chat_with_deonai(user_input):
-    # Simple prompt
-    prompt = f"You are DeonAI, an assistant who knows everything about Chinese. Bio:\n{bio}\n\nUser: {user_input}\nDeonAI:"
-    
-    # Use a lightweight model like deepseek-chat or mistral
-    from transformers import pipeline
-    generator = pipeline("text-generation", model="deepseek-ai/deepseek-llm-7b-chat", max_new_tokens=150)
-    response = generator(prompt)[0]["generated_text"].split("DeonAI:")[-1]
-    
-    return response.strip()
-
-gr.ChatInterface(chat_with_deonai, title="Talk to DeonAI", description="Ask anything about Chinese").launch()
+# Define allowed topics
+ALLOWED_TOPICS = [
+    "ai", "machine learning", "ml", "data science", "blockchain", "web3",
+    "chinese", "turing project", "deonai", "rust", "llm", "llms", "financai", "eucossa"
+]
+
+def is_on_topic(message: str):
+    message = message.lower()
+    return any(topic in message for topic in ALLOWED_TOPICS)
+
+# Define DeonAI logic
+def chat_with_deonai(message, history):
+    if not is_on_topic(message):
+        return "I'm DeonAI! I'm specialized in Chinese’s work, AI, ML, and Web3. Please ask me something in those areas. 😊"
+
+    prompt = f"""
+You are DeonAI, a creative, helpful AI assistant embedded in a portfolio website.
+
+You ONLY discuss topics related to:
+- Chinese (the portfolio owner)
+- His biography and projects
+- AI, Machine Learning, LLMs, Blockchain, and Web3
+- Ongoing projects like Turing Project, DeonAI, FinancAI, etc.
+
+Here is your background info:
+{bio}
+
+Now respond helpfully, concisely, and informatively.
+
+User: {message}
+DeonAI:
+"""
+    try:
+        response = model.generate_content(prompt)
+        return response.text.strip()
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Gradio chat interface
+gr.ChatInterface(
+    fn=chat_with_deonai,
+    title="🤖 DeonAI",
+    description="Ask me anything about Chinese, AI, ML, or Web3! I'm your smart assistant 👨🏽‍💻",
+).launch()