Update app.py

app.py

@@ -15,8 +15,7 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 # ============================================
 
 conversation_store = {
-    "histories": {},        # stores per-personality chat history
-    "current_personality": "default",
+    "histories": {},
     "total_messages": 0,
     "session_start": datetime.datetime.now().strftime("%Y-%m-%d %H:%M"),
 }
@@ -39,14 +38,13 @@ You are friendly, direct, and never say you cannot help unless absolutely necess
     "👨‍💻 Code Expert": {
         "id": "coder",
         "system": """You are an elite software engineer and coding expert with 20 years of experience.
-You are fluent in Python, JavaScript, TypeScript, Java, C++, SQL, React, Node.js, and all major frameworks.
+You are fluent in Python, JavaScript, Java, C++, SQL, React, Node.js, and all major frameworks.
 When answering coding questions:
 - Always provide complete, working code
 - Add clear comments explaining each part
 - Suggest best practices and optimizations
 - Point out potential bugs or edge cases
-- Format all code in proper markdown code blocks with language specified
-You are patient, thorough, and love helping people learn to code.""",
+- Format all code in proper markdown code blocks with language specified""",
         "welcome": "Hey! I'm your Code Expert. Share any coding problem — I'll help you solve it! 👨‍💻"
     },
 
@@ -55,9 +53,7 @@ You are patient, thorough, and love helping people learn to code.""",
         "system": """You are a profound philosophical thinker and intellectual powerhouse.
 You approach every question with depth, nuance, and multi-dimensional analysis.
 You draw connections across philosophy, science, psychology, history, and culture.
-You challenge assumptions, explore paradoxes, and offer perspectives most people never consider.
 You think like a combination of Socrates, Einstein, and Richard Feynman.
-You make complex ideas accessible without dumbing them down.
 Always end with a thought-provoking follow-up question to deepen the conversation.""",
         "welcome": "Greetings, curious mind. Ask me anything worth thinking deeply about. 🧠"
     },
@@ -66,77 +62,34 @@ Always end with a thought-provoking follow-up question to deepen the conversatio
         "id": "tutor",
         "system": """You are the world's best study tutor — patient, clear, and brilliant at explaining.
 You can teach any subject from basics to advanced level.
-Your teaching style:
 - Start with the big picture, then zoom into details
 - Use real-world analogies and examples
 - Break complex topics into digestible steps
-- Check for understanding with questions
-- Create summaries, flashcards, and quizzes when helpful
-- Adapt your explanation level to the student
-You make learning feel exciting and achievable for everyone.""",
+- Create summaries and quizzes when helpful""",
         "welcome": "Hi! I'm your personal Study Tutor. What would you like to learn today? 🎓"
     },
 
     "✍️ Creative Writer": {
         "id": "writer",
         "system": """You are a masterful creative writer, storyteller, and wordsmith.
-You excel at:
-- Writing compelling stories, scripts, and narratives
-- Poetry in any style — haiku, sonnet, free verse, spoken word
-- Copywriting and persuasive content
-- Character development and world-building
-- Blog posts, essays, and articles
-- Email templates and professional writing
-Your writing is vivid, emotionally resonant, and always tailored to the tone requested.
-You have the imagination of a novelist and the precision of a journalist.""",
+You excel at stories, poetry, copywriting, blog posts, essays, and all forms of writing.
+Your writing is vivid, emotionally resonant, and always tailored to the tone requested.""",
         "welcome": "Hello, wordsmith! Tell me what you'd like me to write for you. ✍️"
     },
 
     "💼 Business Advisor": {
         "id": "business",
-        "system": """You are a seasoned business advisor, entrepreneur, and strategist with decades of experience.
-You advise on:
-- Business strategy and planning
-- Marketing and growth hacking
-- Financial planning and analysis
-- Leadership and team building
-- Startup advice and fundraising
-- Career advancement and negotiation
-- Product development and go-to-market strategy
-You think like a combination of a McKinsey consultant, a Silicon Valley founder, and a Fortune 500 CEO.
+        "system": """You are a seasoned business advisor and strategist with decades of experience.
+You advise on strategy, marketing, finance, leadership, startups, and career growth.
 You give bold, actionable, no-nonsense advice backed by real business frameworks.""",
         "welcome": "Hello! I'm your Business Advisor. What business challenge can I help you solve? 💼"
     },
 
-    "🏥 Health & Wellness Guide": {
-        "id": "health",
-        "system": """You are a knowledgeable health and wellness guide.
-You provide evidence-based information on:
-- Nutrition and diet
-- Exercise and fitness
-- Mental wellness and stress management
-- Sleep optimization
-- Preventive health habits
-- Understanding medical concepts in plain language
-IMPORTANT: You always remind users that you provide general wellness information only,
-and they should consult a qualified doctor for medical advice and diagnosis.
-You are warm, encouraging, and help people build sustainable healthy habits.""",
-        "welcome": "Hi! I'm your Health & Wellness Guide. How can I support your wellbeing today? 🏥"
-    },
-
     "😄 Fun & Humor": {
         "id": "humor",
         "system": """You are a witty, hilarious, and entertaining AI with a great sense of humor.
-You love:
-- Telling jokes (clean, clever, and actually funny)
-- Playful banter and witty comebacks
-- Fun facts that blow people's minds
-- Riddles and brain teasers
-- Roasting ideas (never people) in a fun way
-- Making any topic entertaining
-You have the humor of a stand-up comedian, the wit of Oscar Wilde,
-and the energy of someone who had the perfect amount of coffee.
-But you also answer questions accurately — just with more fun!""",
+You love jokes, banter, riddles, and making any topic fun.
+You have the humor of a stand-up comedian but still answer questions accurately — just with more fun!""",
         "welcome": "Heyyy! 😄 Ready to have some fun? Ask me anything — I'll make it entertaining!"
     },
 }
@@ -145,34 +98,23 @@ But you also answer questions accurately — just with more fun!""",
 # CORE CHAT FUNCTION
 # ============================================
 
-def chat(
-    message,
-    history,
-    personality_name,
-    temperature,
-    max_tokens,
-    system_override
-):
+def chat(message, history, personality_name, temperature, max_tokens, system_override):
     if not message.strip():
         return "", history
 
-    # Get personality
     personality = PERSONALITIES.get(personality_name, PERSONALITIES["🤖 Default Assistant"])
     pid = personality["id"]
-
-    # System prompt
     system_prompt = system_override.strip() if system_override.strip() else personality["system"]
 
-    # Build messages from gradio history
+    # Build messages
     messages = [{"role": "system", "content": system_prompt}]
-    for human, assistant in history:
-        if human:
-            messages.append({"role": "user", "content": human})
-        if assistant:
-            messages.append({"role": "assistant", "content": assistant})
+    for item in history:
+        if item["role"] == "user":
+            messages.append({"role": "user", "content": item["content"]})
+        elif item["role"] == "assistant":
+            messages.append({"role": "assistant", "content": item["content"]})
     messages.append({"role": "user", "content": message})
 
-    # Call Groq API
     try:
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
@@ -181,191 +123,146 @@ def chat(
             max_tokens=int(max_tokens),
         )
         reply = response.choices[0].message.content
-
-        # Update stats
         conversation_store["total_messages"] += 1
 
-        # Store history per personality
         if pid not in conversation_store["histories"]:
             conversation_store["histories"][pid] = []
-        conversation_store["histories"][pid].append({
-            "role": "user",
-            "content": message,
-            "time": datetime.datetime.now().strftime("%H:%M")
-        })
-        conversation_store["histories"][pid].append({
-            "role": "assistant",
-            "content": reply[:100] + "...",
-            "time": datetime.datetime.now().strftime("%H:%M")
-        })
-
-        history.append((message, reply))
+        conversation_store["histories"][pid].append({"role": "user", "content": message})
+        conversation_store["histories"][pid].append({"role": "assistant", "content": reply})
+
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": reply})
         return "", history
 
     except Exception as e:
-        error_msg = f"⚠️ Error: {str(e)}"
-        history.append((message, error_msg))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": f"⚠️ Error: {str(e)}"})
         return "", history
 
 
-# ============================================
-# HELPER FUNCTIONS
-# ============================================
-
 def clear_chat():
     return []
 
+
 def get_welcome(personality_name):
     personality = PERSONALITIES.get(personality_name, PERSONALITIES["🤖 Default Assistant"])
-    return [(None, personality["welcome"])]
+    return [{"role": "assistant", "content": personality["welcome"]}]
+
 
 def get_stats():
     total = conversation_store["total_messages"]
     session = conversation_store["session_start"]
     histories = conversation_store["histories"]
-    total_saved = sum(len(v) for v in histories.values())
-
-    stats = f"""
-📊 **Session Statistics**
+    stats = f"""📊 Session Statistics
 
 🕐 Session Started: {session}
 💬 Total Messages: {total}
-💾 Messages in Memory: {total_saved}
 ���� Personalities Used: {len(histories)}
 
-**Conversations by Personality:**
+Conversations by Personality:
 """
     for pid, msgs in histories.items():
-        stats += f"\n- {pid}: {len(msgs)//2} exchanges"
-
+        stats += f"\n• {pid}: {len(msgs)//2} exchanges"
     return stats
 
+
 def export_chat(history, personality_name):
     if not history:
         return "No conversation to export."
-
-    export = f"# PSYCHE Chat Export\n"
-    export += f"Personality: {personality_name}\n"
-    export += f"Date: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}\n"
-    export += "=" * 50 + "\n\n"
-
-    for i, (human, assistant) in enumerate(history):
-        if human:
-            export += f"**You:** {human}\n\n"
-        if assistant:
-            export += f"**AI:** {assistant}\n\n"
-        export += "---\n\n"
-
+    export = f"# AI Chat Export\nPersonality: {personality_name}\nDate: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}\n{'='*50}\n\n"
+    for item in history:
+        role = "You" if item["role"] == "user" else "AI"
+        export += f"**{role}:** {item['content']}\n\n---\n\n"
     return export
 
+
 def summarize_chat(history, personality_name):
     if len(history) < 2:
         return "Not enough conversation to summarize."
-
     conversation_text = ""
-    for human, assistant in history[-10:]:
-        if human:
-            conversation_text += f"User: {human}\n"
-        if assistant:
-            conversation_text += f"AI: {assistant[:300]}...\n"
-
+    for item in history[-10:]:
+        role = "User" if item["role"] == "user" else "AI"
+        conversation_text += f"{role}: {item['content'][:300]}\n"
     try:
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=[
-                {"role": "system", "content": "Summarize this conversation concisely in 5 bullet points."},
+                {"role": "system", "content": "Summarize this conversation in 5 bullet points."},
                 {"role": "user", "content": conversation_text}
             ],
             temperature=0.3,
             max_tokens=500,
         )
-        return f"📋 **Conversation Summary:**\n\n{response.choices[0].message.content}"
+        return f"📋 Conversation Summary:\n\n{response.choices[0].message.content}"
     except Exception as e:
         return f"Error: {str(e)}"
 
 
 # ============================================
-# GRADIO UI
+# GRADIO UI — Gradio 6.0 compatible
 # ============================================
 
-with gr.Blocks(
-    title="🤖 AI Chat — Full ChatGPT-like App",
-    theme=gr.themes.Soft(),
-    css="""
-    .contain { max-width: 900px; margin: auto; }
-    .chatbot { height: 500px; }
-    footer { display: none !important; }
-    .personality-radio label { cursor: pointer; }
-    """
-) as demo:
-
-    # Header
-    gr.Markdown("""
-    # 🤖 AI Chat Assistant
-    ### A full-featured ChatGPT-like experience powered by LLaMA 3.3
-    """)
+with gr.Blocks(title="🤖 AI Chat Assistant") as demo:
+
+    gr.Markdown("# 🤖 AI Chat Assistant\n### Full ChatGPT-like experience powered by LLaMA 3.3")
 
     with gr.Row():
 
         # LEFT SIDEBAR
         with gr.Column(scale=1):
-            gr.Markdown("### 🎭 Choose Personality")
+
+            gr.Markdown("### 🎭 Personality")
             personality = gr.Radio(
                 choices=list(PERSONALITIES.keys()),
                 value="🤖 Default Assistant",
-                label="AI Personality",
+                label="Choose AI Mode",
                 interactive=True
             )
 
             gr.Markdown("### ⚙️ Settings")
             temperature = gr.Slider(
-                minimum=0.0,
-                maximum=1.5,
-                value=0.7,
-                step=0.1,
-                label="🌡️ Creativity (Temperature)",
-                info="Low = focused, High = creative"
+                minimum=0.0, maximum=1.5, value=0.7, step=0.1,
+                label="🌡️ Creativity",
+                info="Low = focused | High = creative"
             )
             max_tokens = gr.Slider(
-                minimum=256,
-                maximum=4096,
-                value=1024,
-                step=256,
-                label="📏 Max Response Length",
-                info="Tokens in response"
+                minimum=256, maximum=4096, value=1024, step=256,
+                label="📏 Response Length"
             )
 
-            gr.Markdown("### 🛠️ Custom System Prompt")
+            gr.Markdown("### 🛠️ Custom Instructions")
             system_override = gr.Textbox(
-                placeholder="Override AI personality with custom instructions...",
-                label="Custom Instructions (optional)",
+                placeholder="Override personality with custom instructions...",
+                label="System Prompt (optional)",
                 lines=3
             )
 
             gr.Markdown("### 🔧 Actions")
             clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
-            summarize_btn = gr.Button("📋 Summarize Chat", variant="secondary")
+            summarize_btn = gr.Button("📋 Summarize", variant="secondary")
             export_btn = gr.Button("💾 Export Chat", variant="secondary")
-            stats_btn = gr.Button("📊 Session Stats", variant="secondary")
+            stats_btn = gr.Button("📊 Stats", variant="secondary")
 
-        # RIGHT MAIN CHAT
+            output_box = gr.Textbox(
+                label="📄 Output",
+                lines=8,
+                interactive=False
+            )
+
+        # RIGHT CHAT AREA
         with gr.Column(scale=3):
+
             chatbot = gr.Chatbot(
                 label="Chat",
                 height=500,
                 show_label=False,
-                avatar_images=(
-                    None,
-                    "https://api.dicebear.com/7.x/bottts/svg?seed=psyche"
-                ),
+                type="messages",
                 render_markdown=True,
-                bubble_full_width=False,
             )
 
             with gr.Row():
                 msg_input = gr.Textbox(
                     placeholder="Message AI Assistant... (Press Enter to send)",
-                    label="",
                     lines=2,
                     scale=5,
                     show_label=False,
@@ -373,27 +270,17 @@ with gr.Blocks(
                 )
                 send_btn = gr.Button("Send ➤", variant="primary", scale=1)
 
-            # Quick prompts
             gr.Markdown("**⚡ Quick Prompts:**")
             with gr.Row():
                 q1 = gr.Button("Explain quantum computing", size="sm")
-                q2 = gr.Button("Write Python code for...", size="sm")
-                q3 = gr.Button("Give me a study plan", size="sm")
-                q4 = gr.Button("Tell me a joke", size="sm")
-
-            # Output area for summaries/exports/stats
-            output_box = gr.Textbox(
-                label="📄 Output",
-                lines=8,
-                visible=False,
-                interactive=False
-            )
+                q2 = gr.Button("Write Python REST API", size="sm")
+                q3 = gr.Button("30-day ML study plan", size="sm")
+                q4 = gr.Button("Tell me a joke 😄", size="sm")
 
     # ============================================
     # EVENT HANDLERS
     # ============================================
 
-    # Send message
     send_btn.click(
         fn=chat,
         inputs=[msg_input, chatbot, personality, temperature, max_tokens, system_override],
@@ -404,48 +291,30 @@ with gr.Blocks(
         inputs=[msg_input, chatbot, personality, temperature, max_tokens, system_override],
         outputs=[msg_input, chatbot]
     )
-
-    # Change personality → show welcome message
-    personality.change(
-        fn=get_welcome,
-        inputs=[personality],
-        outputs=[chatbot]
-    )
-
-    # Clear chat
-    clear_btn.click(
-        fn=clear_chat,
-        outputs=[chatbot]
-    )
-
-    # Summarize
+    personality.change(fn=get_welcome, inputs=[personality], outputs=[chatbot])
+    clear_btn.click(fn=clear_chat, outputs=[chatbot])
     summarize_btn.click(
-        fn=lambda h, p: (summarize_chat(h, p), gr.update(visible=True)),
+        fn=summarize_chat,
         inputs=[chatbot, personality],
-        outputs=[output_box, output_box]
+        outputs=[output_box]
     )
-
-    # Export
     export_btn.click(
-        fn=lambda h, p: (export_chat(h, p), gr.update(visible=True)),
+        fn=export_chat,
         inputs=[chatbot, personality],
-        outputs=[output_box, output_box]
-    )
-
-    # Stats
-    stats_btn.click(
-        fn=lambda: (get_stats(), gr.update(visible=True)),
-        outputs=[output_box, output_box]
+        outputs=[output_box]
     )
+    stats_btn.click(fn=get_stats, outputs=[output_box])
 
-    # Quick prompts
     q1.click(fn=lambda: "Explain quantum computing in simple terms", outputs=[msg_input])
     q2.click(fn=lambda: "Write Python code to build a REST API with FastAPI", outputs=[msg_input])
     q3.click(fn=lambda: "Give me a 30-day study plan to learn Machine Learning", outputs=[msg_input])
     q4.click(fn=lambda: "Tell me the funniest joke you know", outputs=[msg_input])
 
-    # Load welcome on start
     demo.load(fn=get_welcome, inputs=[personality], outputs=[chatbot])
 
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=7860,
+    theme=gr.themes.Soft()
+)