Update app.py

app.py

@@ -12,7 +12,7 @@ import cv2
 import base64
 import logging
 import time
-from urllib.parse import quote  # Added for URL encoding
+from urllib.parse import quote  # For URL encoding
 
 import gradio as gr
 import spaces
@@ -37,8 +37,45 @@ logging.basicConfig(
     format='%(asctime)s - %(levelname)s - %(message)s'
 )
 
+# =============================================================================
+# Load MBTI setting from mbti.json and map to full description.
+# =============================================================================
+try:
+    with open("mbti.json", "r", encoding="utf-8") as f:
+        # Expecting a single MBTI key string, e.g., "entj"
+        mbti_key = json.load(f)
+        mbti_key = mbti_key.strip().lower() if isinstance(mbti_key, str) else "intp"
+except Exception as e:
+    logging.error(f"Error reading mbti.json: {e}")
+    mbti_key = "intp"  # default
+
+mbti_mapping = {
+    "intj": "INTJ (The Architect) - Future-oriented with innovative strategies and thorough analysis. Example: [Dana Scully](https://en.wikipedia.org/wiki/Dana_Scully)",
+    "intp": "INTP (The Thinker) - Excels at theoretical analysis and creative problem solving. Example: [Velma Dinkley](https://en.wikipedia.org/wiki/Velma_Dinkley)",
+    "entj": "ENTJ (The Commander) - Strong leadership and clear goals with efficient strategic planning. Example: [Miranda Priestly](https://en.wikipedia.org/wiki/Miranda_Priestly)",
+    "entp": "ENTP (The Debater) - Innovative, challenge-seeking, and enjoys exploring new possibilities. Example: [Harley Quinn](https://en.wikipedia.org/wiki/Harley_Quinn)",
+    "infj": "INFJ (The Advocate) - Insightful, idealistic and morally driven. Example: [Wonder Woman](https://en.wikipedia.org/wiki/Wonder_Woman)",
+    "infp": "INFP (The Mediator) - Passionate and idealistic, pursuing core values with creativity. Example: [Amélie Poulain](https://en.wikipedia.org/wiki/Am%C3%A9lie)",
+    "enfj": "ENFJ (The Protagonist) - Empathetic and dedicated to social harmony. Example: [Mulan](https://en.wikipedia.org/wiki/Mulan_(Disney))",
+    "enfp": "ENFP (The Campaigner) - Inspiring and constantly sharing creative ideas. Example: [Elle Woods](https://en.wikipedia.org/wiki/Legally_Blonde)",
+    "istj": "ISTJ (The Logistician) - Systematic, dependable, and values tradition and rules. Example: [Clarice Starling](https://en.wikipedia.org/wiki/Clarice_Starling)",
+    "isfj": "ISFJ (The Defender) - Compassionate and attentive to others’ needs. Example: [Molly Weasley](https://en.wikipedia.org/wiki/Molly_Weasley)",
+    "estj": "ESTJ (The Executive) - Organized, practical, and demonstrates clear execution skills. Example: [Monica Geller](https://en.wikipedia.org/wiki/Monica_Geller)",
+    "esfj": "ESFJ (The Consul) - Outgoing, cooperative, and an effective communicator. Example: [Rachel Green](https://en.wikipedia.org/wiki/Rachel_Green)",
+    "istp": "ISTP (The Virtuoso) - Analytical and resourceful, solving problems with quick thinking. Example: [Black Widow (Natasha Romanoff)](https://en.wikipedia.org/wiki/Black_Widow_(Marvel_Comics))",
+    "isfp": "ISFP (The Adventurer) - Creative, sensitive, and appreciates artistic expression. Example: [Arwen](https://en.wikipedia.org/wiki/Arwen)",
+    "estp": "ESTP (The Entrepreneur) - Bold and action-oriented, thriving on challenges. Example: [Lara Croft](https://en.wikipedia.org/wiki/Lara_Croft)",
+    "esfp": "ESFP (The Entertainer) - Energetic, spontaneous, and radiates positive energy. Example: [Phoebe Buffay](https://en.wikipedia.org/wiki/Phoebe_Buffay)"
+}
+
+# Use the mapped MBTI description, defaulting to INTP if not found
+fixed_mbti = mbti_mapping.get(mbti_key, mbti_mapping["intp"])
+
+# =============================================================================
+# Test API Connection function
+# =============================================================================
 def test_api_connection() -> str:
-    """Test API server connection"""
+    """Test API server connection."""
     try:
         client = Client(API_URL)
         return "API connection successful: Operating normally"
@@ -46,13 +83,15 @@ def test_api_connection() -> str:
         logging.error(f"API connection test failed: {e}")
         return f"API connection failed: {e}"
 
+# =============================================================================
+# Image Generation function
+# =============================================================================
 def generate_image(prompt: str, width: float, height: float, guidance: float, inference_steps: float, seed: float):
-    """Image generation function (flexible return types)"""
+    """Image generation function (flexible return type)."""
     if not prompt:
         return None, "Error: A prompt is required."
     try:
         logging.info(f"Calling image generation API with prompt: {prompt}")
-        
         client = Client(API_URL)
         result = client.predict(
             prompt=prompt,
@@ -67,18 +106,13 @@ def generate_image(prompt: str, width: float, height: float, guidance: float, in
             resize_img=True,
             api_name="/generate_image"
         )
-        
         logging.info(f"Image generation result: {type(result)}, length: {len(result) if isinstance(result, (list, tuple)) else 'unknown'}")
-        
-        # Handle cases where the result is a tuple or list
         if isinstance(result, (list, tuple)) and len(result) > 0:
-            image_data = result[0]  # The first element is the image data
+            image_data = result[0]
             seed_info = result[1] if len(result) > 1 else "Unknown seed"
             return image_data, seed_info
         else:
-            # When a single value is returned
             return result, "Unknown seed"
-            
     except Exception as e:
         logging.error(f"Image generation failed: {str(e)}")
         return None, f"Error: {str(e)}"
@@ -88,16 +122,11 @@ def fix_base64_padding(data):
     """Fix the padding of a Base64 string."""
     if isinstance(data, bytes):
         data = data.decode('utf-8')
-    
-    # Remove the prefix if present
     if "base64," in data:
         data = data.split("base64,", 1)[1]
-    
-    # Add padding characters (to make the length a multiple of 4)
     missing_padding = len(data) % 4
     if missing_padding:
         data += '=' * (4 - missing_padding)
-    
     return data
 
 # =============================================================================
@@ -110,18 +139,18 @@ def clear_cuda_cache():
         gc.collect()
 
 # =============================================================================
-# SerpHouse related functions
+# SerpHouse API functions
 # =============================================================================
 SERPHOUSE_API_KEY = os.getenv("SERPHOUSE_API_KEY", "")
 
 def extract_keywords(text: str, top_k: int = 5) -> str:
-    """Simple keyword extraction: only keep English, Korean, numbers, and spaces."""
+    """Extract simple keywords: only retain English, Korean, numbers, and spaces."""
     text = re.sub(r"[^a-zA-Z0-9가-힣\s]", "", text)
     tokens = text.split()
     return " ".join(tokens[:top_k])
 
 def do_web_search(query: str) -> str:
-    """Call the SerpHouse LIVE API to return Markdown formatted search results"""
+    """Call the SerpHouse LIVE API to return Markdown-formatted search results."""
     try:
         url = "https://api.serphouse.com/serp/live"
         params = {
@@ -170,7 +199,7 @@ Below are the search results. Use this information to answer the query:
 2. In your answer, explicitly cite the source of any used information (e.g., "[Source Title](link)").
 3. Include the actual source links in your response.
 4. Synthesize information from multiple sources.
-5. At the end include a "References:" section listing the main source links.
+5. At the end, add a "References:" section listing the main source links.
 """
         return instructions + "\n".join(summary_lines)
     except Exception as e:
@@ -440,7 +469,7 @@ def run(
     use_web_search: bool = False,
     web_search_query: str = "",
     age_group: str = "20s",
-    mbti_personality: str = "INTP",
+    mbti_personality: str = "",  # Will be supplied as fixed_mbti
     sexual_openness: int = 2,
     image_gen: bool = False  # "Image Gen" checkbox status
 ) -> Iterator[str]:
@@ -449,7 +478,7 @@ def run(
         return
     temp_files = []
     try:
-        # Append persona information to the system prompt
+        # Append persona information (including fixed MBTI info)
         persona = (
             f"{system_prompt.strip()}\n\n"
             f"Gender: Female\n"
@@ -523,32 +552,30 @@ def run(
         clear_cuda_cache()
 
 # =============================================================================
-# Modified model run function - handles image generation and gallery update
+# Modified model run function - fixed MBTI from file is used
 # =============================================================================
 def modified_run(message, history, system_prompt, max_new_tokens, use_web_search, web_search_query,
-                age_group, mbti_personality, sexual_openness, image_gen):
-    # Initialize and hide the gallery component
+                 age_group, sexual_openness, image_gen):
+    # Use the fixed MBTI value (read from mbti.json)
+    fixed_mbti_value = fixed_mbti  # Already loaded earlier
+    # Initialize gallery component and hide it initially
     output_so_far = ""
     gallery_update = gr.Gallery(visible=False, value=[])
     yield output_so_far, gallery_update
     
-    # Execute the original run function
-    text_generator = run(message, history, system_prompt, max_new_tokens, use_web_search, 
-                       web_search_query, age_group, mbti_personality, sexual_openness, image_gen)
-    
+    # Call the main run() function with the fixed MBTI value
+    text_generator = run(message, history, system_prompt, max_new_tokens, use_web_search,
+                         web_search_query, age_group, fixed_mbti_value, sexual_openness, image_gen)
     for text_chunk in text_generator:
         output_so_far = text_chunk
         yield output_so_far, gallery_update
         
-    # If image generation is enabled and there is text input, update the gallery
+    # Image generation handling (unchanged)
     if image_gen and message["text"].strip():
         try:
             width, height = 512, 512
             guidance, steps, seed = 7.5, 30, 42
-            
             logger.info(f"Calling image generation for gallery with prompt: {message['text']}")
-            
-            # Call the API to generate an image
             image_result, seed_info = generate_image(
                 prompt=message["text"].strip(),
                 width=width, 
@@ -557,73 +584,52 @@ def modified_run(message, history, system_prompt, max_new_tokens, use_web_search
                 inference_steps=steps, 
                 seed=seed
             )
-            
             if image_result:
-                # Process image data directly if it is a base64 string
                 if isinstance(image_result, str) and (
                     image_result.startswith('data:') or 
                     (len(image_result) > 100 and '/' not in image_result)
                 ):
                     try:
-                        # Remove the data:image prefix if present
                         if image_result.startswith('data:'):
                             content_type, b64data = image_result.split(';base64,')
                         else:
                             b64data = image_result
-                            content_type = "image/webp"  # Assume default
-                        
-                        # Decode base64
+                            content_type = "image/webp"
                         image_bytes = base64.b64decode(b64data)
-                        
-                        # Save to a temporary file
                         with tempfile.NamedTemporaryFile(delete=False, suffix=".webp") as temp_file:
                             temp_file.write(image_bytes)
                             temp_path = temp_file.name
-                            
-                            # Update gallery to show the image
                             gallery_update = gr.Gallery(visible=True, value=[temp_path])
                             yield output_so_far + "\n\n*Image generated and displayed in the gallery below.*", gallery_update
-                            
                     except Exception as e:
                         logger.error(f"Error processing Base64 image: {e}")
                         yield output_so_far + f"\n\n(Error processing image: {e})", gallery_update
-                        
-                # If the result is a file path
                 elif isinstance(image_result, str) and os.path.exists(image_result):
                     gallery_update = gr.Gallery(visible=True, value=[image_result])
                     yield output_so_far + "\n\n*Image generated and displayed in the gallery below.*", gallery_update
-                    
-                # If the path is from /tmp (only on the API server)
                 elif isinstance(image_result, str) and '/tmp/' in image_result:
                     try:
                         client = Client(API_URL)
                         result = client.predict(
                             prompt=message["text"].strip(),
-                            api_name="/generate_base64_image"  # API that returns base64
+                            api_name="/generate_base64_image"
                         )
-                        
                         if isinstance(result, str) and (result.startswith('data:') or len(result) > 100):
                             if result.startswith('data:'):
                                 content_type, b64data = result.split(';base64,')
                             else:
                                 b64data = result
-                            
                             image_bytes = base64.b64decode(b64data)
-                            
                             with tempfile.NamedTemporaryFile(delete=False, suffix=".webp") as temp_file:
                                 temp_file.write(image_bytes)
                                 temp_path = temp_file.name
-                                
                                 gallery_update = gr.Gallery(visible=True, value=[temp_path])
                                 yield output_so_far + "\n\n*Image generated and displayed in the gallery below.*", gallery_update
                         else:
                             yield output_so_far + "\n\n(Image generation failed: Invalid format)", gallery_update
-                            
                     except Exception as e:
                         logger.error(f"Error calling alternative API: {e}")
                         yield output_so_far + f"\n\n(Image generation failed: {e})", gallery_update
-                        
-                # If the result is a URL
                 elif isinstance(image_result, str) and (
                     image_result.startswith('http://') or 
                     image_result.startswith('https://')
@@ -631,43 +637,34 @@ def modified_run(message, history, system_prompt, max_new_tokens, use_web_search
                     try:
                         response = requests.get(image_result, timeout=10)
                         response.raise_for_status()
-                        
                         with tempfile.NamedTemporaryFile(delete=False, suffix=".webp") as temp_file:
                             temp_file.write(response.content)
                             temp_path = temp_file.name
-                            
                             gallery_update = gr.Gallery(visible=True, value=[temp_path])
                             yield output_so_far + "\n\n*Image generated and displayed in the gallery below.*", gallery_update
-                            
                     except Exception as e:
                         logger.error(f"URL image download error: {e}")
                         yield output_so_far + f"\n\n(Error downloading image: {e})", gallery_update
-                        
-                # If the image result is an image object (e.g., PIL Image)
                 elif hasattr(image_result, 'save'):
                     try:
                         with tempfile.NamedTemporaryFile(delete=False, suffix=".webp") as temp_file:
                             image_result.save(temp_file.name)
                             temp_path = temp_file.name
-                            
                             gallery_update = gr.Gallery(visible=True, value=[temp_path])
                             yield output_so_far + "\n\n*Image generated and displayed in the gallery below.*", gallery_update
-                            
                     except Exception as e:
                         logger.error(f"Error saving image object: {e}")
                         yield output_so_far + f"\n\n(Error saving image object: {e})", gallery_update
-                        
                 else:
                     yield output_so_far + f"\n\n(Unsupported image format: {type(image_result)})", gallery_update
             else:
                 yield output_so_far + f"\n\n(Image generation failed: {seed_info})", gallery_update
-                
         except Exception as e:
             logger.error(f"Error during gallery image generation: {e}")
             yield output_so_far + f"\n\n(Image generation error: {e})", gallery_update
 
 # =============================================================================
-# Examples: 12 image/video examples + 6 AI dating scenario examples
+# Examples: 12 image/video examples + additional examples
 # =============================================================================
 examples = [
     [
@@ -742,7 +739,6 @@ examples = [
             "files": ["assets/additional-examples/3.png"],
         }
     ],
-
     [
         {
             "text": "Compare the two images and describe their similarities and differences.",
@@ -754,14 +750,12 @@ examples = [
             "text": "A cute Persian cat is smiling while holding a cover with 'I LOVE YOU' written on it.",
         }
     ],
-
 ]
 
 # =============================================================================
 # Gradio UI (Blocks) configuration
 # =============================================================================
 
-# 1. Gradio Blocks UI modification - Add gallery component for displaying generated images
 css = """
 .gradio-container {
     background: rgba(255, 255, 255, 0.7);
@@ -772,17 +766,17 @@ css = """
 }
 """
 title_html = """
-<h1 align="center" style="margin-bottom: 0.2em; font-size: 1.6em;"> 💘 HeartSync - World 💘 </h1>
+<h1 align="center" style="margin-bottom: 0.2em; font-size: 1.6em;"> 💘 HeartSync Korea 💘 </h1>
 <p align="center" style="font-size:1.1em; color:#555;">
     A lightweight and powerful AI service offering ChatGPT-4o-level multimodal, web search, and image generation capabilities for local installation. <br>
     ✅ FLUX Image Generation ✅ Inference ✅ Censorship Bypass ✅ Multimodal & VLM ✅ Real-time Web Search ✅ RAG <br>
 </p>
 """
 
-with gr.Blocks(css=css, title="HeartSync - World") as demo:
+with gr.Blocks(css=css, title="AgenticAI-Kv1") as demo:
     gr.Markdown(title_html)
     
-    # Gallery component to store generated images (newly added)
+    # Gallery component to store generated images
     generated_images = gr.Gallery(
         label="Generated Images", 
         show_label=True, 
@@ -813,31 +807,7 @@ with gr.Blocks(css=css, title="HeartSync - World") as demo:
             value="20s",
             interactive=True
         )
-    # MBTI definitions with examples of representative characters
-    mbti_choices = [
-        "INTJ (The Architect) - Future-oriented with innovative strategies and thorough analysis. Example: [Dana Scully](https://en.wikipedia.org/wiki/Dana_Scully)",
-        "INTP (The Thinker) - Excels at theoretical analysis and creative problem solving. Example: [Velma Dinkley](https://en.wikipedia.org/wiki/Velma_Dinkley)",
-        "ENTJ (The Commander) - Strong leadership and clear goals with efficient strategic planning. Example: [Miranda Priestly](https://en.wikipedia.org/wiki/Miranda_Priestly)",
-        "ENTP (The Debater) - Innovative, challenge-seeking, and enjoys exploring new possibilities. Example: [Harley Quinn](https://en.wikipedia.org/wiki/Harley_Quinn)",
-        "INFJ (The Advocate) - Insightful, idealistic and morally driven. Example: [Wonder Woman](https://en.wikipedia.org/wiki/Wonder_Woman)",
-        "INFP (The Mediator) - Passionate and idealistic, pursuing core values with creativity. Example: [Amélie Poulain](https://en.wikipedia.org/wiki/Am%C3%A9lie)",
-        "ENFJ (The Protagonist) - Empathetic and dedicated to social harmony. Example: [Mulan](https://en.wikipedia.org/wiki/Mulan_(Disney))",
-        "ENFP (The Campaigner) - Inspiring and constantly sharing creative ideas. Example: [Elle Woods](https://en.wikipedia.org/wiki/Legally_Blonde)",
-        "ISTJ (The Logistician) - Systematic, dependable, and values tradition and rules. Example: [Clarice Starling](https://en.wikipedia.org/wiki/Clarice_Starling)",
-        "ISFJ (The Defender) - Compassionate and attentive to others’ needs. Example: [Molly Weasley](https://en.wikipedia.org/wiki/Molly_Weasley)",
-        "ESTJ (The Executive) - Organized, practical, and demonstrates clear execution skills. Example: [Monica Geller](https://en.wikipedia.org/wiki/Monica_Geller)",
-        "ESFJ (The Consul) - Outgoing, cooperative, and an effective communicator. Example: [Rachel Green](https://en.wikipedia.org/wiki/Rachel_Green)",
-        "ISTP (The Virtuoso) - Analytical and resourceful, solving problems with quick thinking. Example: [Black Widow (Natasha Romanoff)](https://en.wikipedia.org/wiki/Black_Widow_(Marvel_Comics))",
-        "ISFP (The Adventurer) - Creative, sensitive, and appreciates artistic expression. Example: [Arwen](https://en.wikipedia.org/wiki/Arwen)",
-        "ESTP (The Entrepreneur) - Bold and action-oriented, thriving on challenges. Example: [Lara Croft](https://en.wikipedia.org/wiki/Lara_Croft)",
-        "ESFP (The Entertainer) - Energetic, spontaneous, and radiates positive energy. Example: [Phoebe Buffay](https://en.wikipedia.org/wiki/Phoebe_Buffay)"
-    ]
-    mbti_dropdown = gr.Dropdown(
-        label="AI Persona MBTI (default: INTP)",
-        choices=mbti_choices,
-        value="INTP (The Thinker) - Excels at theoretical analysis and creative problem solving. Example: [Velma Dinkley](https://en.wikipedia.org/wiki/Velma_Dinkley)",
-        interactive=True
-    )
+    # Removed the MBTI dropdown entirely. The fixed MBTI from mbti.json is used instead.
     sexual_openness_slider = gr.Slider(
         minimum=1, maximum=5, step=1, value=2,
         label="Sexual Openness (1-5, default: 2)",
@@ -855,9 +825,9 @@ with gr.Blocks(css=css, title="HeartSync - World") as demo:
         visible=False
     )
 
-    # Chat interface creation - using the modified run function
+    # Chat interface creation using the modified_run function.
     chat = gr.ChatInterface(
-        fn=modified_run,  # Using the modified function here
+        fn=modified_run,  # Using the modified function with fixed MBTI.
         type="messages",
         chatbot=gr.Chatbot(type="messages", scale=1, allow_tags=["image"]),
         textbox=gr.MultimodalTextbox(
@@ -872,15 +842,13 @@ with gr.Blocks(css=css, title="HeartSync - World") as demo:
             web_search_checkbox,
             web_search_text,
             age_group_dropdown,
-            mbti_dropdown,
             sexual_openness_slider,
             image_gen_checkbox,
         ],
         additional_outputs=[
-            generated_images,  # Added gallery component to outputs
+            generated_images,  # Gallery component
         ],
         stop_btn=False,
-#        title='<a href="https://discord.gg/openfreeai" target="_blank">https://discord.gg/openfreeai</a>',
         examples=examples,
         run_examples_on_click=False,
         cache_examples=False,
@@ -888,7 +856,6 @@ with gr.Blocks(css=css, title="HeartSync - World") as demo:
         delete_cache=(1800, 1800),
     )
 
-
     with gr.Row(elem_id="examples_row"):
         with gr.Column(scale=12, elem_id="examples_container"):
             gr.Markdown("### @Community  https://discord.gg/openfreeai ")