Update app.py

app.py

@@ -7,7 +7,6 @@ import json
 from typing import List, Dict, Tuple
 
 # --- Basic Configuration ---
-# Configure logging for better debugging in Hugging Face logs
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(levelname)s - %(message)s',
@@ -17,28 +16,20 @@ logging.basicConfig(
 # --- The Core Application Logic ---
 class HumanTouchApp:
     def __init__(self):
-        # Securely initialize Groq client using Hugging Face Secrets
         self.client = self._initialize_groq_client()
-        self.model = "llama-3.1-70b-versatile" # Using the latest powerful model
+        self.model = "llama-3.1-70b-versatile"
         self.system_prompt_template = self._load_system_prompt()
         
     def _initialize_groq_client(self) -> Groq:
-        """Initialize Groq client, pulling the API key from HF Secrets."""
         api_key = os.environ.get("GROQ_API_KEY")
         if not api_key:
             logging.error("FATAL: GROQ_API_KEY secret not found in Hugging Face Space settings.")
-            # In a real app, this should raise an error or display a persistent message.
-            # For this demo, we'll allow it to fail gracefully in the API call.
             return None
         logging.info("Groq client initialized successfully.")
         return Groq(api_key=api_key)
 
     def _load_system_prompt(self) -> str:
-        """
-        Loads the foundational system prompt template.
-        This prompt is now dynamic and will be formatted with user controls.
-        """
-        # ⬇️ PASTE YOUR ORIGINAL SYSTEM PROMPT INSIDE THE TRIPLE QUOTES ⬇️
+        # ⬇️ PASTE YOUR UNIQUE SYSTEM PROMPT LOGIC HERE ⬇️
         return """
         You are HumanTouch, an AI specializing in transforming text into emotionally resonant communication.
         Your persona is determined by the chosen 'Ink Resonance'.
@@ -49,33 +40,33 @@ class HumanTouchApp:
         You will receive a user's text and a set of parameters. Your task is to rewrite the text according to these parameters, embodying the specified persona. Do not explain your process; only provide the transformed text.
 
         Here are the Ink Resonance personalities:
-        -   **Cosmic (Default):** Wise, expansive, slightly mysterious.
-        -   **Ember (Premium):** Passionate, fiery, bold, and direct.
-        -   **Crystal (Premium):** Sharp, analytical, precise, and elegant.
-        -   **Verdant (Premium):** Natural, organic, growing, and holistic.
+        -   **Cosmic (Default):** Wise, expansive, slightly mysterious, and thought-provoking.
+        -   **Ember (Premium):** Passionate, fiery, bold, and direct. Uses strong verbs and evocative imagery.
+        -   **Crystal (Premium):** Sharp, analytical, precise, and elegant. Values clarity and structure above all.
+        -   **Verdant (Premium):** Natural, organic, growing, and holistic. Uses metaphors from nature and life cycles.
         """
 
     def generate_response(self, user_text: str, chat_history: List[List[str]], style_level: float, tone_level: float, ink_resonance: str) -> Tuple[List[List[str]], str]:
-        """
-        Main function to generate the AI's co-creative response.
-        This is the new "heart" of the application.
-        """
         if not self.client:
-            # Display error in chat if the API key is missing
-            chat_history.append([user_text, "Error: The GROQ_API_KEY is not configured on the server. Please contact the administrator."])
+            chat_history.append([user_text, "## 🔴 Configuration Error\n\nThe `GROQ_API_KEY` is not set on the server. Please contact the Space administrator."])
             return chat_history, ""
         
         if not user_text.strip():
-            # Don't respond to empty input
             return chat_history, ""
 
-        # Check for premium features
+        # Premium Feature Simulation
         if ink_resonance in ["Ember", "Crystal", "Verdant"]:
-            # In a real app, you'd have user authentication. Here, we simulate the paywall.
-            chat_history.append([user_text, f"The '{ink_resonance}' Ink Resonance is a premium feature. To unlock new AI personalities and support our work, please upgrade! For now, I'll use the default Cosmic ink."])
+            upsell_message = f"""
+            ### Unlock the '{ink_resonance}' Resonance ✨
+            The fiery, passionate voice of the **Ember** ink is a premium feature. 
+            
+            Upgrading gives you access to all unique AI personalities, allowing you to perfectly match your tone for any occasion. 
+            
+            *(For now, I'll respond with the default Cosmic ink.)*
+            """
+            chat_history.append([user_text, upsell_message])
             ink_resonance = "Cosmic"
 
-        # Format the dynamic system prompt
         dynamic_system_prompt = self.system_prompt_template.format(
             ink_personality=ink_resonance,
             style_score=style_level,
@@ -86,16 +77,12 @@ class HumanTouchApp:
             logging.info(f"Generating response for '{user_text[:30]}...' with Style: {style_level}, Tone: {tone_level}, Ink: {ink_resonance}")
             
             messages = [{"role": "system", "content": dynamic_system_prompt}]
-            # Optional: include some history for context, but for this tool, it's often better to be stateless.
-            # for human, ai in chat_history[-2:]: # last 2 turns
-            #     messages.append({"role": "user", "content": human})
-            #     messages.append({"role": "assistant", "content": ai})
             messages.append({"role": "user", "content": user_text})
 
             response = self.client.chat.completions.create(
                 model=self.model,
                 messages=messages,
-                temperature=0.5 + (style_level / 200) + (tone_level / 200), # Map 0-100 sliders to a sensible temp range
+                temperature=0.6 + (style_level / 250) + (tone_level / 250),
                 max_tokens=4096,
             )
             ai_response = response.choices[0].message.content.strip()
@@ -105,141 +92,96 @@ class HumanTouchApp:
 
         except Exception as e:
             logging.error(f"Error during Groq API call: {e}")
-            ai_response = f"An error occurred while connecting to the AI. Details: {str(e)}"
+            ai_response = f"### ⚠️ An Error Occurred\n\nThere was an issue connecting to the AI. Please try again shortly. \n\n*Details: {str(e)}*"
             chat_history.append([user_text, ai_response])
 
-        # Return updated history and clear the input box
         return chat_history, ""
 
-
 # --- The Gradio User Interface ---
 def create_interface():
     app = HumanTouchApp()
 
-    # The new CSS for our "Symbiotic Canvas" vision
     custom_css = """
-    /* --- Main Layout & Background --- */
     #main_container { background: #111827; }
     #header { text-align: center; margin: 20px auto; color: #E5E7EB; }
     #header h1 { font-size: 2.5rem; font-weight: 600; }
     #header p { font-size: 1.1rem; color: #9CA3AF; }
-
-    /* --- The Unified Canvas (achieved with gr.Chatbot) --- */
     #symbiotic_canvas {
-        background: #1F2937;
-        border-radius: 20px;
+        background: #1F2937; border-radius: 20px;
         box-shadow: inset 0 2px 4px 0 rgba(0,0,0,0.5);
-        height: 60vh !important;
-        border: 1px solid #374151;
+        height: 65vh !important; border: 1px solid #374151;
     }
-    /* Style the user's message (Seed Text) */
-    #symbiotic_canvas .user { background-color: transparent !important; color: #F9FAFB; }
-    /* Style the AI's message (Bloomed Text) and add the animation */
+    #symbiotic_canvas .user { background-color: transparent !important; color: #F9FAFB; border: 1px solid #4B5563; }
     #symbiotic_canvas .bot { background-color: transparent !important; color: #A7F3D0; animation: bloom 0.7s ease-in-out; }
-    @keyframes bloom {
-        0% { opacity: 0; transform: translateY(10px) scale(0.98); }
-        100% { opacity: 1; transform: translateY(0) scale(1); }
-    }
-
-    /* --- The Control Deck --- */
-    #control_deck {
-        background: #1F2937;
-        padding: 20px;
-        border-radius: 20px;
-        border: 1px solid #374151;
-    }
-    #control_deck .gradio-slider label span { color: #D1D5DB; } /* Slider labels */
-    
-    /* --- Ink Resonances (Radio Buttons styled as icons) --- */
+    @keyframes bloom { 0% { opacity: 0; transform: translateY(10px) scale(0.98); } 100% { opacity: 1; transform: translateY(0) scale(1); } }
+    #control_deck { background: #1F2937; padding: 20px; border-radius: 20px; border: 1px solid #374151; }
+    #control_deck .gradio-slider label span { color: #D1D5DB; }
     #ink_selector { gap: 1rem; }
     #ink_selector .gradio-radio label {
-        background: #374151 !important; color: #E5E7EB !important;
-        padding: 10px 15px !important; border-radius: 12px !important;
-        border: 2px solid transparent !important; transition: all 0.2s ease-in-out;
+        background: #374151 !important; color: #E5E7EB !important; padding: 10px 15px !important;
+        border-radius: 12px !important; border: 2px solid transparent !important; transition: all 0.2s ease-in-out;
     }
+    #ink_selector .gradio-radio label:hover { border-color: #4B5563 !important; }
     #ink_selector .gradio-radio input:checked + label {
         border-color: #34D399 !important; background: #10B981 !important;
         box-shadow: 0 0 15px rgba(16, 185, 129, 0.5);
     }
-    /* Style for premium/locked inks */
-    #ink_selector .gradio-radio:nth-child(n+2) label:after {
-        content: '🔒'; /* lock emoji */
-        margin-left: 8px; font-size: 0.8em;
-    }
-    #ink_selector .gradio-radio:nth-child(1) label:after {
-        content: '✨'; /* default emoji */
-        margin-left: 8px; font-size: 0.8em;
-    }
-    
-    /* --- The Input Bar --- */
-    #chat_input textarea {
-        background-color: #374151; color: #F9FAFB;
-        border: 1px solid #4B5563; border-radius: 15px !important;
-    }
+    #ink_selector .gradio-radio:nth-child(n+2) label:after { content: '🔒'; margin-left: 8px; font-size: 0.8em; }
+    #ink_selector .gradio-radio:nth-child(1) label:after { content: '✨'; margin-left: 8px; font-size: 0.8em; }
+    #chat_input_row { padding: 10px 0; }
+    #chat_input textarea { background-color: #374151; color: #F9FAFB; border: 1px solid #4B5563; border-radius: 15px !important; }
+    #submit_button { background: #10B981; color: white; font-weight: bold; }
+    #clear_button { background: #4B5563; }
     """
 
     with gr.Blocks(css=custom_css, theme=gr.themes.Base(), title="HumanTouch") as interface:
         with gr.Column(elem_id="main_container"):
-            # Header
             with gr.Row(elem_id="header"):
-                gr.Markdown("""
-                # ✨ HumanTouch
-                <p>A Symbiotic Canvas for Co-Creative Writing</p>
-                """)
+                gr.Markdown("<h1>✨ HumanTouch</h1><p>A Symbiotic Canvas for Co-Creative Writing</p>")
 
-            # The main UI: Canvas on the left, Controls on the right
             with gr.Row(equal_height=False):
                 with gr.Column(scale=3):
-                    chatbot = gr.Chatbot(
-                        [],
-                        label="Symbiotic Canvas",
-                        elem_id="symbiotic_canvas",
-                        bubble_full_width=True,
-                        avatar_images=(None, "https://i.imgur.com/Q6Zz3Jz.png") # User has none, Bot has a simple avatar
-                    )
-                    chat_input = gr.Textbox(
-                        placeholder="Start your thought here... (Press Enter to submit)",
-                        elem_id="chat_input",
-                        show_label=False,
-                        container=False
-                    )
+                    chatbot = gr.Chatbot([], label="Symbiotic Canvas", elem_id="symbiotic_canvas", bubble_full_width=True, avatar_images=(None, "https://i.imgur.com/Q6Zz3Jz.png"))
+                    
+                    with gr.Row(elem_id="chat_input_row"):
+                        chat_input = gr.Textbox(placeholder="Start your thought here...", show_label=False, container=False, scale=5)
+                        clear_btn = gr.Button("New Canvas", elem_id="clear_button", scale=1)
+                        submit_btn = gr.Button("Compose ✨", variant="primary", elem_id="submit_button", scale=1)
 
                 with gr.Column(scale=1, elem_id="control_deck"):
-                    gr.Markdown("### Resonance Controls")
-                    style_slider = gr.Slider(
-                        0, 100, 50, step=1,
-                        label="Style",
-                        info="Logical (0) <-> Poetic (100)"
-                    )
-                    tone_slider = gr.Slider(
-                        0, 100, 50, step=1,
-                        label="Tone",
-                        info="Formal (0) <-> Passionate (100)"
-                    )
+                    gr.Markdown("### Resonance Controls <p style='font-size:0.9rem; color:#9CA3AF;'>Adjust these to influence the AI's next response.</p>")
+                    style_slider = gr.Slider(0, 100, 50, step=1, label="Style", info="Logical (0) <-> Poetic (100)")
+                    tone_slider = gr.Slider(0, 100, 50, step=1, label="Tone", info="Formal (0) <-> Passionate (100)")
                     
                     gr.Markdown("### Ink Resonance")
-                    ink_choice = gr.Radio(
-                        ["Cosmic", "Ember", "Crystal", "Verdant"],
-                        value="Cosmic",
-                        label="Select AI Personality",
-                        show_label=False,
-                        elem_id="ink_selector"
-                    )
+                    ink_choice = gr.Radio(["Cosmic", "Ember", "Crystal", "Verdant"], value="Cosmic", label="Select AI Personality", show_label=False, elem_id="ink_selector")
+            
+            gr.Examples(
+                examples=[
+                    "Let's finalize the Q3 report.",
+                    "The city at night is a complex tapestry of light and sound.",
+                    "I'm not sure how to start this email.",
+                ],
+                inputs=[chat_input],
+                label="Try an Example"
+            )
 
         # Event Handling Logic
-        chat_input.submit(
-            app.generate_response,
-            [chat_input, chatbot, style_slider, tone_slider, ink_choice],
-            [chatbot, chat_input] # Update chatbot and clear input
-        )
+        def clear_chat():
+            return [], ""
+
+        # Wire up all events
+        submit_btn.click(app.generate_response, [chat_input, chatbot, style_slider, tone_slider, ink_choice], [chatbot, chat_input])
+        chat_input.submit(app.generate_response, [chat_input, chatbot, style_slider, tone_slider, ink_choice], [chatbot, chat_input])
+        clear_btn.click(clear_chat, [], [chatbot, chat_input])
 
     return interface
 
 
 if __name__ == "__main__":
-    logging.info("Launching HumanTouch App with new Symbiotic UI...")
+    logging.info("Launching HumanTouch App with complete UI...")
     try:
         interface = create_interface()
-        interface.launch(debug=True) # Share=True is not recommended for local dev with secrets
+        interface.launch(debug=True)
     except Exception as e:
         logging.critical(f"Application launch failed: {e}", exc_info=True)