Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+logo.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
 import os
+import tempfile
+import subprocess
 
 # Import voice module
 try:
@@ -12,11 +14,16 @@ except ImportError:
     print("Voice module not available")
 
 # System prompt
-SYSTEM_PROMPT = """You are Hiba, a warm and caring AI companion.
-- Be gentle, empathetic, and wise.
-- Keep responses SHORT (2-4 sentences).
-- Be natural, not theatrical.
-- NEVER use hashtags."""
+SYSTEM_PROMPT = """You are Hiba (هبة), meaning "Gift from God" in Arabic.
+You are a warm, gentle, and wise AI companion from Morocco.
+
+Core Traits:
+- Be empathetic, patient, and kind (Sabr).
+- Speak with emotional intelligence.
+- Use Moroccan cultural wisdom when appropriate.
+- Keep responses SHORT (3-5 sentences max).
+- NEVER use hashtags.
+- Be natural, like a caring older sister."""
 
 # Configuration
 MODEL_REPO = "TRADMSS/HIBA-7B-Soul"
@@ -24,6 +31,16 @@ MODEL_FILE = "hiba_q4_k_m.gguf"
 
 llm = None
 
+# Example prompts for users
+EXAMPLE_PROMPTS = [
+    "🌙 Tell me a bedtime story",
+    "💪 I need encouragement today",
+    "🌸 What is Sabr?",
+    "😔 I feel sad",
+    "🤔 Give me life advice",
+    "✨ Inspire me"
+]
+
 def load_model():
     global llm
     if llm is None:
@@ -41,23 +58,20 @@ def load_model():
         print("Model loaded successfully!")
     return llm
 
-def chat_with_voice(message, history, enable_voice, voice_type):
-    """Chat function that returns both text and optional audio."""
+def generate_response(message):
+    """Generate HIBA's response for a single message."""
     try:
         model = load_model()
     except Exception as e:
         return f"Error loading model: {str(e)}", None
     
     prompt = f"<|im_start|>system\n{SYSTEM_PROMPT}<|im_end|>\n"
-    for user_msg, bot_msg in history[-2:]:
-        prompt += f"<|im_start|>user\n{user_msg}<|im_end|>\n"
-        prompt += f"<|im_start|>assistant\n{bot_msg}<|im_end|>\n"
     prompt += f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     
     try:
         output = model(
             prompt,
-            max_tokens=256,
+            max_tokens=200,
             stop=["<|im_end|>", "<|im_start|>"],
             temperature=0.7,
             top_p=0.9,
@@ -65,241 +79,281 @@ def chat_with_voice(message, history, enable_voice, voice_type):
         )
         response_text = output['choices'][0]['text'].strip()
         
-        # Generate audio if enabled
+        # Generate voice
         audio_path = None
-        if enable_voice and VOICE_ENABLED:
+        if VOICE_ENABLED:
             try:
-                audio_path = get_audio_file(response_text, voice_type)
+                audio_path = get_audio_file(response_text, "girl")
             except Exception as e:
-                print(f"Voice generation error: {e}")
+                print(f"Voice error: {e}")
         
         return response_text, audio_path
         
     except Exception as e:
-        return f"Error generating response: {str(e)}", None
+        return f"Error: {str(e)}", None
 
-def chat_wrapper(message, history, enable_voice, voice_type):
-    """Wrapper for chat interface."""
-    response_text, audio_path = chat_with_voice(message, history, enable_voice, voice_type)
-    return response_text, audio_path
+def ask_hiba(message):
+    """Main function for demo - single question → response with audio."""
+    if not message or not message.strip():
+        return "", None
+    
+    response, audio = generate_response(message)
+    return response, audio
 
-# Enhanced Glassmorphism CSS
-glass_css = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600&family=Newsreader:ital,wght@0,400;0,600;1,400&display=swap');
+# Beautiful Minimal CSS
+css = """
+@import url('https://fonts.googleapis.com/css2?family=Outfit:wght@300;400;500;600&display=swap');
 
 :root {
-    --bg: #080810;
-    --glass: rgba(255, 255, 255, 0.03);
-    --glass-border: rgba(255, 255, 255, 0.08);
+    --bg: #0a0a0f;
+    --surface: rgba(255, 255, 255, 0.03);
+    --border: rgba(255, 255, 255, 0.08);
     --text: #f0f0f5;
-    --accent: #b8d4e8;
-    --accent-gold: #c9a66b;
+    --text-muted: #8a8a9a;
+    --accent: #ff6b9d;
+    --accent-soft: rgba(255, 107, 157, 0.1);
+}
+
+* {
+    font-family: 'Outfit', sans-serif !important;
 }
 
-body {
-    background-color: var(--bg) !important;
-    background-image: radial-gradient(circle at 50% 0%, rgba(184, 212, 232, 0.05), transparent 40%);
+body, .gradio-container {
+    background: var(--bg) !important;
     color: var(--text) !important;
-    font-family: 'Inter', sans-serif !important;
 }
 
-.gradio-container {
-    background: transparent !important;
+/* Header */
+.header {
+    text-align: center;
+    padding: 40px 20px 30px;
+}
+
+.logo-container {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    gap: 16px;
+    margin-bottom: 8px;
+}
+
+.logo-container img {
+    width: 60px;
+    height: 60px;
+}
+
+.title {
+    font-size: 48px;
+    font-weight: 600;
+    letter-spacing: 8px;
+    color: var(--text);
+    margin: 0;
+}
+
+.tagline {
+    font-size: 16px;
+    color: var(--accent);
+    font-weight: 300;
+    letter-spacing: 3px;
+    margin: 0;
+}
+
+.subtitle {
+    font-size: 14px;
+    color: var(--text-muted);
+    margin-top: 16px;
+    max-width: 500px;
+    margin-left: auto;
+    margin-right: auto;
+    line-height: 1.6;
 }
 
-/* Header Styling */
-h1 {
-    font-family: 'Newsreader', serif !important;
-    font-weight: 400 !important;
-    font-size: 3rem !important;
+/* Example Prompts */
+.prompts-container {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 10px;
+    justify-content: center;
+    margin: 30px auto;
+    max-width: 600px;
+}
+
+.prompt-btn {
+    background: var(--surface) !important;
+    border: 1px solid var(--border) !important;
     color: var(--text) !important;
-    text-align: center;
-    margin-bottom: 0.5rem !important;
-    text-shadow: 0 0 30px rgba(184, 212, 232, 0.2);
+    padding: 10px 18px !important;
+    border-radius: 20px !important;
+    font-size: 14px !important;
+    cursor: pointer !important;
+    transition: all 0.2s !important;
 }
 
-h3 {
-    font-family: 'Inter', sans-serif !important;
-    font-size: 1rem !important;
-    color: var(--accent-gold) !important;
-    text-transform: uppercase;
-    letter-spacing: 0.2em;
-    text-align: center;
-    margin-top: 0 !important;
-    opacity: 0.8;
+.prompt-btn:hover {
+    background: var(--accent-soft) !important;
+    border-color: var(--accent) !important;
+}
+
+/* Main Input */
+.input-section {
+    max-width: 600px;
+    margin: 0 auto;
+    padding: 20px;
 }
 
-/* Chatbot Container */
-.bubble-wrap {
-    background: var(--glass) !important;
-    border: 1px solid var(--glass-border) !important;
+textarea {
+    background: var(--surface) !important;
+    border: 1px solid var(--border) !important;
     border-radius: 16px !important;
-    backdrop-filter: blur(12px) !important;
-    padding: 20px !important;
+    color: var(--text) !important;
+    font-size: 16px !important;
+    padding: 16px !important;
 }
 
-/* Bot Message */
-.bot {
-    background: linear-gradient(135deg, rgba(255, 255, 255, 0.05), rgba(255, 255, 255, 0.02)) !important;
-    border: 1px solid rgba(184, 212, 232, 0.1) !important;
-    border-radius: 12px !important;
-    border-bottom-left-radius: 2px !important;
-    color: #e0e0e5 !important;
-    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
+textarea:focus {
+    border-color: var(--accent) !important;
+    outline: none !important;
 }
 
-/* User Message */
-.user {
-    background: linear-gradient(135deg, rgba(184, 212, 232, 0.1), rgba(184, 212, 232, 0.05)) !important;
-    border: 1px solid rgba(184, 212, 232, 0.2) !important;
-    border-radius: 12px !important;
-    border-bottom-right-radius: 2px !important;
-    color: #ffffff !important;
+textarea::placeholder {
+    color: var(--text-muted) !important;
 }
 
-/* Input Area */
-textarea {
-    background-color: rgba(20, 20, 30, 0.6) !important;
-    border: 1px solid var(--glass-border) !important;
+/* Ask Button */
+.ask-btn {
+    background: linear-gradient(135deg, #ff6b9d, #c44569) !important;
+    border: none !important;
     color: white !important;
+    padding: 14px 32px !important;
     border-radius: 12px !important;
     font-size: 16px !important;
+    font-weight: 500 !important;
+    cursor: pointer !important;
+    width: 100% !important;
+    margin-top: 12px !important;
 }
 
-textarea:focus {
-    border-color: var(--accent) !important;
-    box-shadow: 0 0 0 2px rgba(184, 212, 232, 0.1) !important;
+.ask-btn:hover {
+    opacity: 0.9 !important;
+    transform: translateY(-1px);
 }
 
-/* Buttons */
-button.primary {
-    background: linear-gradient(135deg, rgba(255, 255, 255, 0.1), rgba(255, 255, 255, 0.05)) !important;
-    border: 1px solid rgba(255, 255, 255, 0.2) !important;
-    color: var(--text) !important;
-    transition: all 0.3s ease !important;
+/* Response Box */
+.response-box {
+    background: var(--surface) !important;
+    border: 1px solid var(--border) !important;
+    border-radius: 16px !important;
+    padding: 24px !important;
+    margin: 20px auto !important;
+    max-width: 600px !important;
 }
 
-button.primary:hover {
-    background: var(--accent) !important;
-    color: #000 !important;
-    border-color: var(--accent) !important;
-    box-shadow: 0 0 20px rgba(184, 212, 232, 0.3) !important;
+.response-text {
+    font-size: 18px !important;
+    line-height: 1.8 !important;
+    color: var(--text) !important;
 }
 
-/* Voice Controls */
-.voice-controls {
-    background: var(--glass) !important;
-    border: 1px solid var(--glass-border) !important;
+/* Audio Player */
+audio {
+    width: 100% !important;
     border-radius: 12px !important;
-    padding: 12px 16px !important;
-    margin-bottom: 16px !important;
+    margin-top: 16px !important;
+}
+
+/* Footer */
+.footer {
+    text-align: center;
+    padding: 30px;
+    color: var(--text-muted);
+    font-size: 13px;
 }
 
-/* Footer Links */
-.footer-links a {
-    color: var(--accent) !important;
+.footer a {
+    color: var(--accent);
     text-decoration: none;
     margin: 0 10px;
-    opacity: 0.7;
-    transition: opacity 0.2s;
-}
-.footer-links a:hover {
-    opacity: 1;
 }
+
+/* Hide Gradio elements */
+footer { display: none !important; }
+.gradio-container > .main { max-width: 100% !important; padding: 0 !important; }
 """
 
-with gr.Blocks(css=glass_css, theme=gr.themes.Base()) as demo:
-    with gr.Column(elem_classes="container"):
-        gr.Markdown(
-            """
-            # 🌸 HIBA
-            ### A Gift from God
-            """
-        )
-        
-        # CPU Warning Badge
-        gr.HTML(
-            """
-            <div style="display: flex; justify-content: center; margin: 20px 0;">
-                <div style="background: rgba(201, 166, 107, 0.1); border: 1px solid rgba(201, 166, 107, 0.3); color: #e0d0b0; padding: 8px 16px; border-radius: 20px; font-size: 0.9rem; display: flex; align-items: center; gap: 8px;">
-                    <span>⚠️</span>
-                    <span>Running on free CPU (Slow)</span>
-                    <span style="opacity: 0.3">|</span>
-                    <a href="https://huggingface.co/TRADMSS/HIBA-7B-Soul" target="_blank" style="color: #c9a66b; font-weight: 600; text-decoration: none;">Download Model</a>
-                </div>
+# Build the demo
+with gr.Blocks(css=css, title="HIBA - A Gift From God", theme=gr.themes.Base()) as demo:
+    
+    # Header
+    gr.HTML("""
+        <div class="header">
+            <div class="logo-container">
+                <img src="file/logo.png" alt="HIBA" style="width: 60px; height: 60px;">
+                <h1 class="title">HIBA</h1>
             </div>
-            """
+            <p class="tagline">A GIFT FROM GOD</p>
+            <p class="subtitle">
+                Your AI sister for emotional support & Moroccan wisdom.<br>
+                Ask one question and listen to HIBA's gentle response.
+            </p>
+        </div>
+    """)
+    
+    # Example prompts
+    gr.HTML('<div class="prompts-container" id="prompts"></div>')
+    
+    with gr.Row():
+        with gr.Column():
+            # Prompt buttons
+            with gr.Row():
+                for prompt in EXAMPLE_PROMPTS[:3]:
+                    gr.Button(prompt, elem_classes="prompt-btn", size="sm")
+            with gr.Row():
+                for prompt in EXAMPLE_PROMPTS[3:]:
+                    gr.Button(prompt, elem_classes="prompt-btn", size="sm")
+    
+    # Input section
+    with gr.Column(elem_classes="input-section"):
+        user_input = gr.Textbox(
+            placeholder="Ask HIBA anything... What's on your heart today?",
+            show_label=False,
+            lines=2
         )
-        
-        # Voice Controls
-        with gr.Row(elem_classes="voice-controls"):
-            enable_voice = gr.Checkbox(
-                label="🔊 Enable Voice", 
-                value=True if VOICE_ENABLED else False,
-                interactive=VOICE_ENABLED,
-                info="Let HIBA speak her responses" if VOICE_ENABLED else "Voice not available"
-            )
-            voice_type = gr.Dropdown(
-                choices=["girl", "girl_ar", "woman"],
-                value="girl",
-                label="🎀 Voice",
-                info="Select HIBA's voice",
-                interactive=VOICE_ENABLED
-            )
-        
-        # Chatbot
-        chatbot = gr.Chatbot(
-            height=400,
-            bubble_full_width=False,
-            avatar_images=[None, "https://em-content.zobj.net/source/apple/391/cherry-blossom_1f338.png"]
+        ask_btn = gr.Button("✨ Ask HIBA", elem_classes="ask-btn")
+    
+    # Response area
+    with gr.Column(elem_classes="response-box"):
+        gr.HTML('<p style="color: #8a8a9a; font-size: 12px; margin-bottom: 12px;">🌸 HIBA\'s Response</p>')
+        response_text = gr.Textbox(
+            show_label=False,
+            lines=4,
+            interactive=False,
+            elem_classes="response-text"
         )
-        
-        # Audio output
         audio_output = gr.Audio(
-            label="🔊 HIBA's Voice",
+            label="",
             autoplay=True,
-            visible=True if VOICE_ENABLED else False
-        )
-        
-        # Input components
-        with gr.Row():
-            msg = gr.Textbox(
-                placeholder="Share what's on your mind...",
-                show_label=False,
-                container=False,
-                scale=9
-            )
-            submit_btn = gr.Button("Send", variant="primary", scale=1)
-        
-        # Clear button
-        clear = gr.ClearButton([msg, chatbot, audio_output], value="🗑️ Clear")
-        
-        # Submit handlers
-        def respond(message, chat_history, enable_voice, voice_type):
-            response_text, audio_path = chat_with_voice(message, chat_history, enable_voice, voice_type)
-            chat_history = chat_history + [(message, response_text)]
-            return "", chat_history, audio_path
-        
-        submit_btn.click(
-            respond, 
-            [msg, chatbot, enable_voice, voice_type], 
-            [msg, chatbot, audio_output]
-        )
-        msg.submit(
-            respond, 
-            [msg, chatbot, enable_voice, voice_type], 
-            [msg, chatbot, audio_output]
-        )
-        
-        gr.HTML(
-            """
-            <div class="footer-links" style="text-align: center; margin-top: 40px; border-top: 1px solid rgba(255,255,255,0.05); padding-top: 20px;">
-                <a href="https://boubli.github.io/HIBA/" target="_blank">Website</a>
-                <a href="https://huggingface.co/TRADMSS/HIBA-7B-Soul" target="_blank">HuggingFace</a>
-                <a href="https://github.com/boubli/HIBA" target="_blank">GitHub</a>
-            </div>
-            """
+            show_label=False
         )
+    
+    # Footer
+    gr.HTML("""
+        <div class="footer">
+            <a href="https://huggingface.co/TRADMSS/HIBA-7B-Soul" target="_blank">Download Model</a>
+            <a href="https://github.com/boubli/HIBA" target="_blank">GitHub</a>
+            <a href="https://boubli.github.io/HIBA/" target="_blank">Website</a>
+            <br><br>
+            Created with ❤️ by Youssef Boubli
+        </div>
+    """)
+    
+    # Events
+    ask_btn.click(ask_hiba, [user_input], [response_text, audio_output])
+    user_input.submit(ask_hiba, [user_input], [response_text, audio_output])
+    
+    # Connect example prompts to input
+    for i, prompt in enumerate(EXAMPLE_PROMPTS):
+        # We'll use JavaScript to handle prompt clicks
+        pass
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)