Update app.py

app.py

@@ -2,99 +2,98 @@ import os
 import logging
 import gradio as gr
 from groq import Groq
-from datetime import datetime
-import json
-from typing import List, Dict, Tuple
+from typing import List, Tuple
 
 # --- Basic Configuration ---
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    handlers=[logging.StreamHandler()]
-)
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', handlers=[logging.StreamHandler()])
 
-# --- The Core Application Logic ---
+# --- The Core Application Logic (Now serves both modes) ---
 class HumanTouchApp:
     def __init__(self):
         self.client = self._initialize_groq_client()
         self.model = "llama-3.3-70b-versatile"
         self.system_prompt_template = self._load_system_prompt()
-        
+
     def _initialize_groq_client(self) -> Groq:
         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.")
+            logging.error("FATAL: GROQ_API_KEY secret not found.")
             return None
         logging.info("Groq client initialized successfully.")
         return Groq(api_key=api_key)
 
     def _load_system_prompt(self) -> str:
-        # ⬇️ PASTE YOUR UNIQUE SYSTEM PROMPT LOGIC HERE ⬇️
+        # This one prompt now powers both modes, using context to differentiate tasks.
         return """
         You are HumanTouch, an AI specializing in transforming text into emotionally resonant communication.
-        Your persona is determined by the chosen 'Ink Resonance'.
-        Your output style is guided by two scores:
-        1.  **Style Score (0-100):** 0 is purely logical and precise. 100 is highly poetic, abstract, and metaphorical.
-        2.  **Tone Score (0-100):** 0 is formal and stoic. 100 is casual, emotional, and passionate.
-
-        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, 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.
-        """
+        Your output style is guided by two scores: Style (0=Logical, 100=Poetic) and Tone (0=Formal, 100=Passionate).
 
-    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]:
-        if not self.client:
-            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():
-            return chat_history, ""
+        You will be given a task: either "HUMANIZE" a block of text or "CO-CREATE" from a user's seed phrase.
 
-        # Premium Feature Simulation
-        if ink_resonance in ["Ember", "Crystal", "Verdant"]:
-            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"
-
-        dynamic_system_prompt = self.system_prompt_template.format(
-            ink_personality=ink_resonance,
-            style_score=style_level,
-            tone_score=tone_level
-        )
+        **Task: HUMANIZE**
+        When given a block of text, your goal is to rewrite it to sound more natural, engaging, and less robotic, according to the provided Style and Tone scores. Retain the core meaning but infuse it with a human touch.
 
-        try:
-            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}]
-            messages.append({"role": "user", "content": user_text})
+        **Task: CO-CREATE**
+        When given a short phrase or idea, your goal is to expand and "bloom" it into a more complete, resonant thought, guided by the Style and Tone scores. This is a creative partnership.
+
+        In both tasks, do not explain your process. Only provide the transformed text.
+        """
 
+    def _call_groq_api(self, messages: List[dict], style_level: float, tone_level: float) -> str:
+        if not self.client:
+            return "## 🔴 Configuration Error\n\nThe `GROQ_API_KEY` is not set on the server. Please contact the Space administrator."
+        try:
+            logging.info(f"Calling Groq API with Style: {style_level}, Tone: {tone_level}")
             response = self.client.chat.completions.create(
                 model=self.model,
                 messages=messages,
                 temperature=0.6 + (style_level / 250) + (tone_level / 250),
                 max_tokens=4096,
             )
-            ai_response = response.choices[0].message.content.strip()
-            
-            chat_history.append([user_text, ai_response])
-            logging.info("Successfully generated response.")
-
+            return response.choices[0].message.content.strip()
         except Exception as e:
             logging.error(f"Error during Groq API call: {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 f"### ⚠️ An Error Occurred\n\nThere was an issue connecting to the AI. Please try again shortly. \n\n*Details: {str(e)}*"
+
+    # Method for the "Humanizer" Tab
+    def humanize_block_text(self, text_to_humanize: str, style_level: float, tone_level: float) -> str:
+        if not text_to_humanize.strip():
+            return "Please paste some AI-generated text to get started."
+
+        user_content = f"""
+        TASK: HUMANIZE
+        Style Score: {style_level}
+        Tone Score: {tone_level}
+        ---
+        TEXT TO HUMANIZE:
+        {text_to_humanize}
+        """
+        messages = [
+            {"role": "system", "content": self.system_prompt_template},
+            {"role": "user", "content": user_content}
+        ]
+        return self._call_groq_api(messages, style_level, tone_level)
+
+    # Method for the "Co-Creative Canvas" Tab
+    def generate_co_creation(self, user_text: str, chat_history: List[List[str]], style_level: float, tone_level: float) -> Tuple[List[List[str]], str]:
+        if not user_text.strip():
+            return chat_history, ""
 
+        user_content = f"""
+        TASK: CO-CREATE
+        Style Score: {style_level}
+        Tone Score: {tone_level}
+        ---
+        SEED PHRASE:
+        {user_text}
+        """
+        messages = [
+            {"role": "system", "content": self.system_prompt_template},
+            {"role": "user", "content": user_content}
+        ]
+        
+        ai_response = self._call_groq_api(messages, style_level, tone_level)
+        chat_history.append([user_text, ai_response])
         return chat_history, ""
 
 # --- The Gradio User Interface ---
@@ -105,81 +104,89 @@ def create_interface():
     #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; }
-    #symbiotic_canvas {
-        background: #1F2937; border-radius: 20px;
-        box-shadow: inset 0 2px 4px 0 rgba(0,0,0,0.5);
-        height: 65vh !important; border: 1px solid #374151;
-    }
+    #header p { font-size: 1.1rem; color: #9CA3AF; margin-bottom: 20px;}
+    .gradio-tabs { background-color: #1F2937; border-radius: 20px !important; border: 1px solid #374151; }
+    .tab-buttons button { background-color: #1F2937 !important; color: #9CA3AF !important; border-radius: 10px 10px 0 0 !important; }
+    .tab-buttons button.selected { background-color: #374151 !important; color: white !important; }
+    
+    /* --- Humanizer Tab Styles --- */
+    #humanizer_input, #humanizer_output { background-color: #374151; color: #F9FAFB; border: 1px solid #4B5563; border-radius: 15px !important; min-height: 50vh;}
+    #humanize_btn { background: #10B981; color: white; font-weight: bold; padding: 15px 0; font-size: 1.2rem; }
+
+    /* --- Canvas Tab Styles --- */
+    #symbiotic_canvas { background: #1F2937; border-radius: 20px; box-shadow: inset 0 2px 4px 0 rgba(0,0,0,0.5); height: 65vh !important; border: 1px solid #374151; }
     #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); } }
     #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;
-    }
-    #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);
-    }
-    #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"):
             with gr.Row(elem_id="header"):
-                gr.Markdown("<h1>✨ HumanTouch</h1><p>A Symbiotic Canvas for Co-Creative Writing</p>")
-
-            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"))
-                    
-                    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 <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")
-            
-            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"
-            )
+                gr.Markdown("<h1>✨ HumanTouch</h1><p>Transform robotic text or co-create with a symbiotic AI partner. Choose your mode below.</p>")
+
+            with gr.Tabs() as tabs:
+                # --- TAB 1: The Humanizer ---
+                with gr.Tab("Humanizer", id="humanizer_tab"):
+                    with gr.Column():
+                        gr.Markdown("### 🤖➡️🧑 Humanize AI Text\n<p>Paste your text, adjust the resonance, and see it transformed.</p>")
+                        with gr.Row():
+                            with gr.Column(scale=2):
+                                input_text_humanizer = gr.Textbox(label="Paste AI Text Here", lines=20, elem_id="humanizer_input")
+                            with gr.Column(scale=2):
+                                output_text_humanizer = gr.Textbox(label="Humanized Result", lines=20, interactive=False, elem_id="humanizer_output")
+                        with gr.Row():
+                            with gr.Column(scale=1):
+                                style_slider_h = gr.Slider(0, 100, 50, label="Style (Logical <-> Poetic)")
+                            with gr.Column(scale=1):
+                                tone_slider_h = gr.Slider(0, 100, 50, label="Tone (Formal <-> Passionate)")
+                        humanize_button = gr.Button("Humanize ✨", variant="primary", elem_id="humanize_btn")
+                        gr.Examples(
+                            [["The system's analysis concluded that the optimal parameters were achieved through iterative processing."],
+                             ["Our team will action the deliverables post-haste to synergize our market position."]],
+                            inputs=[input_text_humanizer], label="Try an Example"
+                        )
+
+                # --- TAB 2: The Co-Creative Canvas ---
+                with gr.Tab("Co-Creative Canvas", id="canvas_tab"):
+                    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"))
+                            with gr.Row(elem_id="chat_input_row"):
+                                chat_input = gr.Textbox(placeholder="Start a thought and press Enter...", show_label=False, container=False, scale=4)
+                                clear_btn = gr.Button("New Canvas", scale=1)
+                        with gr.Column(scale=1, elem_id="control_deck"):
+                            gr.Markdown("### Resonance Controls")
+                            style_slider_c = gr.Slider(0, 100, 50, label="Style", info="Logical <-> Poetic")
+                            tone_slider_c = gr.Slider(0, 100, 50, label="Tone", info="Formal <-> Passionate")
+                            gr.Examples(
+                                ["The city at night", "I'm not sure how to start this email.", "Let's brainstorm names for the new project."],
+                                inputs=[chat_input], label="Try a Seed Phrase"
+                            )
 
         # Event Handling Logic
-        def clear_chat():
-            return [], ""
+        # Humanizer Tab
+        humanize_button.click(
+            app.humanize_block_text,
+            [input_text_humanizer, style_slider_h, tone_slider_h],
+            [output_text_humanizer]
+        )
 
-        # 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])
+        # Canvas Tab
+        chat_input.submit(
+            app.generate_co_creation,
+            [chat_input, chatbot, style_slider_c, tone_slider_c],
+            [chatbot, chat_input]
+        )
+        clear_btn.click(lambda: ([], ""), [], [chatbot, chat_input])
 
     return interface
 
 
 if __name__ == "__main__":
-    logging.info("Launching HumanTouch App with complete UI...")
+    logging.info("Launching HumanTouch App with Blended UI...")
     try:
         interface = create_interface()
         interface.launch(debug=True)