Demo version

app.py

@@ -63,7 +63,7 @@ The code should be complete, functional, and follow Flutter best practices.
         # Additional cleanup for common issues
         code = code.replace("dart\n", "").replace("Dart\n", "")
 
-        return code if code else "// AI-generated placeholder code"
+        return code if code else "// Generated using Qwen 3 0.6B AI model\n// Please try rephrasing your request"
 
     except Exception as e:
         print(f"Error generating code with AI: {e}")
@@ -71,19 +71,18 @@ The code should be complete, functional, and follow Flutter best practices.
 
 def generate_flutter_code(description):
     """
-    Generate Flutter code files based on user description.
-    This is a placeholder implementation that creates sample Dart files.
-    In production, this would call an AI model (Claude, GPT, etc.) via API.
-    
+    Generate Flutter code files based on user description using Qwen 3 0.6B AI model.
+
     Args:
         description: User's description of the Flutter app/widget
-        
+
     Returns:
-        Path to generated ZIP file or None if error
+        Tuple of (file_path, status_message, main_code, widgets_code, models_code, services_code)
     """
     
     if not description or len(description.strip()) < 10:
-        return None, "❌ Please provide a more detailed description (at least 10 characters)."
+        error_message = "❌ Please provide a more detailed description (at least 10 characters)."
+        return None, error_message, "// Waiting for description", "// Waiting for description", "// Waiting for description", "// Waiting for description"
     
     try:
         # Create timestamp for unique filenames
@@ -126,17 +125,18 @@ def generate_flutter_code(description):
             f.write(zip_buffer.getvalue())
         
         success_message = f"""
-✅ **Flutter code generated successfully!**
+✅ **Flutter code generated successfully using Qwen 3 0.6B!**
 
+🤖 **AI Model:** Qwen 3 0.6B (specialized for code generation)
 📝 **Your description:** {description[:100]}{'...' if len(description) > 100 else ''}
 
 📦 **Generated files:**
 - `lib/main.dart` - Main application entry point
-- `lib/widgets.dart` - Custom widgets
-- `lib/models.dart` - Data models
-- `lib/services.dart` - Business logic and services
+- `lib/widgets.dart` - Custom widgets with Material Design
+- `lib/models.dart` - Data models with JSON serialization
+- `lib/services.dart` - Business logic and validation services
 - `pubspec.yaml` - Project dependencies
-- `README.md` - Project documentation
+- `README.md` - Setup and usage instructions
 
 💡 **Next steps:**
 1. Download the ZIP file below
@@ -144,13 +144,14 @@ def generate_flutter_code(description):
 3. Run `flutter pub get` in the project directory
 4. Run `flutter run` to launch the app
 
-⚠️ **Note:** This is AI-generated placeholder code. Review and customize it for your needs.
+⚠️ **Important:** This is AI-generated code using Qwen 3 0.6B. While high-quality, always review and test before production use.
         """
-        
-        return output_path, success_message
+
+        return output_path, success_message, main_dart, widgets_dart, models_dart, services_dart
         
     except Exception as e:
-        return None, f"❌ Error generating Flutter code: {str(e)}"
+        error_message = f"❌ Error generating Flutter code: {str(e)}"
+        return None, error_message, "// Error occurred", "// Error occurred", "// Error occurred", "// Error occurred"
 
 
 def generate_main_dart(description):
@@ -714,7 +715,7 @@ def create_interface():
                     lines=6,
                     max_lines=10
                 )
-                
+
                 gr.Markdown(
                     """
                     **💡 Tips for better results:**
@@ -724,23 +725,57 @@ def create_interface():
                     - Describe any animations or interactions
                     """
                 )
-                
+
                 generate_btn = gr.Button(
-                    "🎨 Generate Flutter Code",
+                    "🎨 Generate Flutter Code with Qwen 3 0.6B",
                     variant="primary",
                     size="lg"
                 )
-            
+
             with gr.Column(scale=1):
                 status_output = gr.Markdown(
                     label="Status",
                     value="👋 Ready to generate your Flutter app!"
                 )
-                
+
                 file_output = gr.File(
                     label="📦 Download Flutter Project",
                     type="filepath"
                 )
+
+        # Code preview tabs
+        with gr.Tabs():
+            with gr.TabItem("📱 main.dart"):
+                main_code_output = gr.Code(
+                    label="Main Application Code",
+                    language="dart",
+                    lines=20,
+                    value="// Click 'Generate Flutter Code' to see your main.dart code here"
+                )
+
+            with gr.TabItem("🎨 widgets.dart"):
+                widgets_code_output = gr.Code(
+                    label="Custom Widgets Code",
+                    language="dart",
+                    lines=20,
+                    value="// Click 'Generate Flutter Code' to see your widgets.dart code here"
+                )
+
+            with gr.TabItem("📊 models.dart"):
+                models_code_output = gr.Code(
+                    label="Data Models Code",
+                    language="dart",
+                    lines=20,
+                    value="// Click 'Generate Flutter Code' to see your models.dart code here"
+                )
+
+            with gr.TabItem("⚙️ services.dart"):
+                services_code_output = gr.Code(
+                    label="Business Logic Code",
+                    language="dart",
+                    lines=20,
+                    value="// Click 'Generate Flutter Code' to see your services.dart code here"
+                )
         
         gr.Markdown(
             """
@@ -774,7 +809,7 @@ def create_interface():
         generate_btn.click(
             fn=generate_flutter_code,
             inputs=[description_input],
-            outputs=[file_output, status_output]
+            outputs=[file_output, status_output, main_code_output, widgets_code_output, models_code_output, services_code_output]
         )
         
         # Add example buttons