Update README.md

README.md

@@ -17,28 +17,34 @@ models:
  - VIDraft/Gemma-3-R1984-4B 
 ---
 
-ChartGPT : AI-Powered Multi Diagram Generator
+ChartGPT : AI-Powered Multi Diagram & Design Generator
 VIDEO Link: https://www.youtube.com/watch?v=erof86WoNMY
 
 <div align="center">
-  <h3>🎨 AI Service that Generates Professional Diagrams from Natural Language</h3>
-  <p>Based LLM 'Gemma-3-R1984-27B' Model. Powered by VIDraft⚡</p>
-  <p><em>Ranked 2nd on Google DeepMind' FACTS Grounding Leaderboard</em></p>
+  <h3>🎨 AI Service that Generates Professional Diagrams and Visual Designs from Natural Language</h3>
+  <p>Based on LLM 'Gemma-3-R1984-27B' Model & FLUX Image Generation Model. Powered by VIDraft⚡</p>
+  <p><em>Ranked 2nd on Google DeepMind FACTS Grounding Leaderboard</em></p>
 </div>
 
-Service Link:  https://huggingface.co/spaces/openfree/Chart-GPT
+Service Link: https://huggingface.co/spaces/openfree/Chart-GPT
 
 📌 Overview
-ChartGPT is an AI-based service that automatically generates professional diagrams from Korean and English prompts. Without needing to learn complex diagramming tools, simply describe what you want and AI will automatically generate structured diagrams.
+ChartGPT is an AI-based service that automatically generates professional diagrams and visual designs from Korean and English prompts. Without needing to learn complex diagramming tools, simply describe what you want and AI will automatically generate structured diagrams or beautiful visual designs.
 
 🚀 Core Technology
-Based LLM: Gemma-3-R1984-27B
+Base LLM: Gemma-3-R1984-27B
 
-Achieved 2nd place on Google DeepMind FACTS Grounding Leaderboard
+Achieved 2nd place on Google DeepMind FACTS Grounding Leaderboard(https://huggingface.co/spaces/MaziyarPanahi/FACTS-Leaderboard)
 State-of-the-art language model with exceptional factuality and accuracy
 Outstanding performance in both Korean and English
 Optimized for complex structuring tasks
 
+Image Generation: FLUX.1-schnell
+
+High-quality image generation based on latest diffusion models
+Fast generation speed with high quality
+Support for various styles and concepts
+
 MCP (Model Context Protocol) Support
 
 Standardized AI Integration: Full support for Anthropic's MCP protocol
@@ -54,19 +60,29 @@ Extensible Architecture: Operates as an MCP server for easy connection with othe
 🔄 Process Flow: Represent workflows or decision-making processes sequentially
 📋 WBS Diagram: Display projects broken down hierarchically
 
-2. AI-Based Automatic Generation
+2. Support for 6 Visual Design Types (NEW)
+
+🎨 Product Design: Industrial design concepts and product sketches
+🧠 Mindmap: Visually rich thought maps
+📱 Mockup: UI/UX wireframes and app designs
+📈 Infographic: Data visualization and information design
+📐 Diagram: Business workflows and process visualization
+📊 Flowchart: Decision flows and process charts
+
+3. AI-Based Automatic Generation
 
 Generate diagrams from natural language prompts
-Support for both Korean and English
+Automatic translation from Korean to English (for Design Types)
 Automatic structured JSON generation
+Optimized prompt templates for each design type
 
-3. Brave Search Integration
+4. Brave Search Integration
 
 Reflect latest information through real-time web search
 Generate more accurate and richer content
 Optional activation
 
-4. Flexible Output Formats
+5. Flexible Output Formats
 
 PNG format (high-quality images)
 SVG format (vector graphics, infinitely scalable)
@@ -81,57 +97,59 @@ Multimodal AI Pipeline: Automate text analysis → diagram generation
 
 📖 How to Use
 Using the AI Assistant Tab
+1. Enter Prompt
 
-Enter Prompt
-
-Describe the diagram you want in natural language
-Example: "Create a flowchart showing the machine learning process"
-
+Describe the diagram or design you want in natural language
+Diagram example: "Create a flowchart showing the machine learning process"
+Design example: "smartphone banking app design"
 
-Select Diagram Type
+2. Select Type
 
-Choose the appropriate diagram type from the dropdown
+Diagram Type: Generate structured diagrams (JSON-based)
+Design Type: Generate visual designs (FLUX-based, English only)
 
-
-Use Brave Search (Optional)
+3. Use Brave Search (Optional)
 
 Activate checkbox when latest information is needed
-Web search results will be reflected in diagram generation
-
-
-Click Generate Button
-
-AI automatically generates JSON structure
-Diagram image automatically rendered
+Web search results will be reflected in generation
 
+4. Click Generate Button
 
+AI automatically generates structure
+Diagram or design image automatically rendered
 
 Direct JSON Input
+
 Fine-tune by directly editing JSON in each diagram type tab
+
 🎯 Use Cases
 Business
 
 Project planning (WBS)
 Business process documentation (Process Flow)
 Organizational chart creation (Radial Diagram)
+Product concept design (Product Design)
+Marketing infographics (Infographic)
 
 Education
 
-Learning content organization (Concept Map)
+Learning content organization (Concept Map, Mindmap)
 Course explanatory materials (Synoptic Chart)
 Research topic structuring (Concept Map)
+Educational material visualization (Diagram)
 
 Technical Documentation
 
 System architecture diagramming
 API flow explanation
 Data processing pipeline visualization
+UI/UX mockups (Mockup)
+Technical flowcharts (Flowchart)
 
 🔧 Advanced Configuration
 Customizing JSON Structure
 Each diagram type has its unique JSON structure:
 Concept Map Example:
-
 json{
   "central_node": "Artificial Intelligence",
   "nodes": [
@@ -143,34 +161,43 @@ json{
     }
   ]
 }
-
 Prompt Writing Tips
 
 Be Specific: "AI classification system" → "AI classification system including deep learning, machine learning, and natural language processing"
 Specify Relationships: "A is part of B", "X leads to Y", etc.
 Describe Hierarchical Structure: "A at the top level, with B and C below", etc.
+For Design Types: Input in Korean is automatically translated to English
 
 💡 Model Performance
 Gemma-3-R1984-27B Features
 
-High Factuality: 2nd place on Google DeepMind FACTS Leaderboard
+High Factuality
 Structuring Ability: Accurately understands complex relationships and hierarchical structures
 Multilingual Support: Ensures equal quality in both Korean and English
 Context Understanding: Accurately grasps prompt intent
 
+FLUX.1-schnell Features
+
+High-Quality Images: Professional-level visual design generation
+Fast Generation: Efficient inference for quick results
+Diverse Styles: Support for sketches, wireframes, infographics, and various styles
+
 🛠️ Tech Stack
 
 Frontend: Gradio (with built-in MCP server)
-AI Model: Gemma-3-R1984-27B (2nd on FACTS Grounding Leaderboard)
+AI Models:
+
+Gemma-3-R1984-27B
+FLUX.1-schnell (High-quality image generation)
+
+
 Diagram Engine: Graphviz
 Search Engine: Brave Search API
 Protocol: MCP (Model Context Protocol)
 Language: Python
 
-
 <div align="center">
   <p>Made with ❤️ by VIDraft Team</p>
   <p><strong>Powered by VIDraft⚡</strong></p>
   <p><em>MCP-enabled for seamless AI integration</em></p>
-</div>
-
+</div>