Demo version

app.py

@@ -3,6 +3,73 @@ import zipfile
 import io
 import os
 from datetime import datetime
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+import torch
+
+# Global model variables
+model = None
+tokenizer = None
+
+def load_model():
+    """Load the small AI model for code generation."""
+    global model, tokenizer
+    if model is None:
+        try:
+            # Use a very small model that can run on Hugging Face Spaces free tier
+            model_name = "distilgpt2"
+
+            print("Loading AI model...")
+            tokenizer = AutoTokenizer.from_pretrained(model_name)
+            model = AutoModelForCausalLM.from_pretrained(model_name)
+
+            # Set padding token to EOS token for GPT-2
+            tokenizer.pad_token = tokenizer.eos_token
+
+            print("Model loaded successfully!")
+        except Exception as e:
+            print(f"Error loading model: {e}")
+            return False
+    return True
+
+def generate_code_with_ai(prompt, max_length=500):
+    """Generate code using the AI model."""
+    if not load_model():
+        return "// Error: Could not load AI model\n// Using fallback template instead"
+
+    try:
+        # Create a more specific prompt for Flutter/Dart code
+        full_prompt = f"Write Flutter Dart code for: {prompt}\n\n"
+
+        inputs = tokenizer(full_prompt, return_tensors="pt", padding=True, truncation=True, max_length=100)
+
+        with torch.no_grad():
+            outputs = model.generate(
+                inputs["input_ids"],
+                attention_mask=inputs["attention_mask"],
+                max_length=max_length,
+                num_return_sequences=1,
+                temperature=0.7,
+                do_sample=True,
+                pad_token_id=tokenizer.eos_token_id,
+                eos_token_id=tokenizer.eos_token_id,
+            )
+
+        generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+        # Extract the code part (remove the prompt)
+        code = generated_text[len(full_prompt):].strip()
+
+        # Basic cleanup - try to extract Dart-like code
+        if "```dart" in code:
+            code = code.split("```dart")[1].split("```")[0]
+        elif "```" in code:
+            code = code.split("```")[1].split("```")[0]
+
+        return code if code else "// AI-generated placeholder code"
+
+    except Exception as e:
+        print(f"Error generating code with AI: {e}")
+        return "// Error generating code\n// Using fallback template"
 
 def generate_flutter_code(description):
     """
@@ -89,20 +156,22 @@ def generate_flutter_code(description):
 
 
 def generate_main_dart(description):
-    """Generate the main.dart file content."""
-    return f"""// AI-Generated Flutter Application
-// Description: {description}
-// Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+    """Generate the main.dart file content using AI."""
+    prompt = f"""Create a Flutter main.dart file for an app described as: {description}
 
+The main.dart should include:
+- Material Design theme
+- A basic home page
+- Proper imports
+- Clean, modern Flutter code structure
+
+```dart
 import 'package:flutter/material.dart';
-import 'widgets.dart';
 
 void main() {{
   runApp(const MyApp());
 }}
 
-/// Main application widget
-/// This is the root of your Flutter application
 class MyApp extends StatelessWidget {{
   const MyApp({{super.key}});
 
@@ -110,22 +179,15 @@ class MyApp extends StatelessWidget {{
   Widget build(BuildContext context) {{
     return MaterialApp(
       title: 'Flutter App',
-      debugShowCheckedModeBanner: false,
       theme: ThemeData(
-        colorScheme: ColorScheme.fromSeed(seedColor: Colors.blue),
+        primarySwatch: Colors.blue,
         useMaterial3: true,
-        appBarTheme: const AppBarTheme(
-          centerTitle: true,
-          elevation: 2,
-        ),
       ),
       home: const HomePage(),
     );
   }}
 }}
 
-/// Home page widget
-/// Main screen of the application
 class HomePage extends StatefulWidget {{
   const HomePage({{super.key}});
 
@@ -134,66 +196,44 @@ class HomePage extends StatefulWidget {{
 }}
 
 class _HomePageState extends State<HomePage> {{
-  int _counter = 0;
-
-  void _incrementCounter() {{
-    setState(() {{
-      _counter++;
-    }});
-  }}
-
   @override
   Widget build(BuildContext context) {{
     return Scaffold(
-      appBar: AppBar(
-        backgroundColor: Theme.of(context).colorScheme.inversePrimary,
-        title: const Text('Flutter App'),
-      ),
-      body: Center(
-        child: Column(
-          mainAxisAlignment: MainAxisAlignment.center,
-          children: <Widget>[
-            // Custom widget from widgets.dart
-            const CustomInfoCard(
-              title: 'Welcome',
-              description: '{description[:50]}...',
-            ),
-            const SizedBox(height: 24),
-            const Text(
-              'You have pushed the button this many times:',
-              style: TextStyle(fontSize: 16),
-            ),
-            Text(
-              '$_counter',
-              style: Theme.of(context).textTheme.headlineMedium?.copyWith(
-                    fontWeight: FontWeight.bold,
-                    color: Theme.of(context).colorScheme.primary,
-                  ),
-            ),
-          ],
-        ),
-      ),
-      floatingActionButton: FloatingActionButton(
-        onPressed: _incrementCounter,
-        tooltip: 'Increment',
-        child: const Icon(Icons.add),
-      ),
+      appBar: AppBar(title: const Text('Home')),
+      body: const Center(child: Text('Hello World')),
     );
   }}
 }}
-"""
+```"""
 
+    ai_generated_code = generate_code_with_ai(prompt, max_length=800)
 
-def generate_widgets_dart(description):
-    """Generate the widgets.dart file content."""
-    return f"""// Custom Widgets for the Flutter Application
+    # Add header comment
+    header = f"""// AI-Generated Flutter Application
 // Description: {description}
 // Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+// Note: This is AI-generated code - review and test before production use
+
+"""
+
+    return header + ai_generated_code
 
+
+def generate_widgets_dart(description):
+    """Generate the widgets.dart file content using AI."""
+    prompt = f"""Create Flutter custom widgets for an app described as: {description}
+
+Create at least 3 reusable widgets including:
+- A custom card/info widget
+- A custom button widget
+- A custom input field widget
+- Follow Material Design guidelines
+- Include proper documentation
+
+```dart
 import 'package:flutter/material.dart';
 
 /// Custom info card widget
-/// Displays a title and description in a styled card
 class CustomInfoCard extends StatelessWidget {{
   final String title;
   final String description;
@@ -216,26 +256,11 @@ class CustomInfoCard extends StatelessWidget {{
         child: Column(
           mainAxisSize: MainAxisSize.min,
           children: [
-            if (icon != null)
-              Icon(
-                icon,
-                size: 48,
-                color: Theme.of(context).colorScheme.primary,
-              ),
+            if (icon != null) Icon(icon, size: 48),
             if (icon != null) const SizedBox(height: 12),
-            Text(
-              title,
-              style: Theme.of(context).textTheme.headlineSmall?.copyWith(
-                    fontWeight: FontWeight.bold,
-                  ),
-              textAlign: TextAlign.center,
-            ),
+            Text(title, style: Theme.of(context).textTheme.headlineSmall),
             const SizedBox(height: 8),
-            Text(
-              description,
-              style: Theme.of(context).textTheme.bodyMedium,
-              textAlign: TextAlign.center,
-            ),
+            Text(description, style: Theme.of(context).textTheme.bodyMedium),
           ],
         ),
       ),
@@ -243,105 +268,65 @@ class CustomInfoCard extends StatelessWidget {{
   }}
 }}
 
-/// Custom button widget with loading state
+/// Custom button with loading state
 class CustomButton extends StatelessWidget {{
   final String text;
   final VoidCallback? onPressed;
   final bool isLoading;
-  final IconData? icon;
 
   const CustomButton({{
     super.key,
     required this.text,
     this.onPressed,
     this.isLoading = false,
-    this.icon,
   }});
 
   @override
   Widget build(BuildContext context) {{
     return ElevatedButton(
       onPressed: isLoading ? null : onPressed,
-      style: ElevatedButton.styleFrom(
-        padding: const EdgeInsets.symmetric(horizontal: 24, vertical: 12),
-        shape: RoundedRectangleBorder(
-          borderRadius: BorderRadius.circular(8),
-        ),
-      ),
       child: isLoading
-          ? const SizedBox(
-              width: 20,
-              height: 20,
-              child: CircularProgressIndicator(strokeWidth: 2),
-            )
-          : Row(
-              mainAxisSize: MainAxisSize.min,
-              children: [
-                if (icon != null) ...[
-                  Icon(icon, size: 20),
-                  const SizedBox(width: 8),
-                ],
-                Text(text),
-              ],
-            ),
+          ? const CircularProgressIndicator()
+          : Text(text),
     );
   }}
 }}
+```"""
 
-/// Custom text input field widget
-class CustomTextField extends StatelessWidget {{
-  final String label;
-  final String? hint;
-  final TextEditingController? controller;
-  final int maxLines;
-  final TextInputType? keyboardType;
+    ai_generated_code = generate_code_with_ai(prompt, max_length=1000)
 
-  const CustomTextField({{
-    super.key,
-    required this.label,
-    this.hint,
-    this.controller,
-    this.maxLines = 1,
-    this.keyboardType,
-  }});
+    # Add header comment
+    header = f"""// Custom Widgets for the Flutter Application
+// Description: {description}
+// Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+// Note: This is AI-generated code - review and test before production use
 
-  @override
-  Widget build(BuildContext context) {{
-    return TextField(
-      controller: controller,
-      maxLines: maxLines,
-      keyboardType: keyboardType,
-      decoration: InputDecoration(
-        labelText: label,
-        hintText: hint,
-        border: OutlineInputBorder(
-          borderRadius: BorderRadius.circular(8),
-        ),
-        filled: true,
-        fillColor: Theme.of(context).colorScheme.surface,
-      ),
-    );
-  }}
-}}
 """
 
+    return header + ai_generated_code
+
 
 def generate_models_dart(description):
-    """Generate the models.dart file content."""
-    return f"""// Data Models for the Flutter Application
-// Description: {description}
-// Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+    """Generate the models.dart file content using AI."""
+    prompt = f"""Create Flutter data models for an app described as: {description}
+
+Create data models with:
+- Proper Dart class structure
+- JSON serialization methods (toJson/fromJson)
+- CopyWith methods for immutability
+- Proper type annotations
+- At least 2-3 relevant models based on the app description
 
+```dart
 /// Base item model
-/// Represents a generic item in the application
 class Item {{
   final String id;
   final String title;
   final String description;
   final DateTime createdAt;
-  bool isCompleted;
+  final bool isCompleted;
 
-  Item({{
+  const Item({{
     required this.id,
     required this.title,
     required this.description,
@@ -349,108 +334,101 @@ class Item {{
     this.isCompleted = false,
   }});
 
-  /// Convert item to JSON
-  Map<String, dynamic> toJson() {{
-    return {{
-      'id': id,
-      'title': title,
-      'description': description,
-      'createdAt': createdAt.toIso8601String(),
-      'isCompleted': isCompleted,
-    }};
-  }}
+  Map<String, dynamic> toJson() => {{
+    'id': id,
+    'title': title,
+    'description': description,
+    'createdAt': createdAt.toIso8601String(),
+    'isCompleted': isCompleted,
+  }};
+
+  factory Item.fromJson(Map<String, dynamic> json) => Item(
+    id: json['id'],
+    title: json['title'],
+    description: json['description'],
+    createdAt: DateTime.parse(json['createdAt']),
+    isCompleted: json['isCompleted'] ?? false,
+  );
 
-  /// Create item from JSON
-  factory Item.fromJson(Map<String, dynamic> json) {{
-    return Item(
-      id: json['id'] as String,
-      title: json['title'] as String,
-      description: json['description'] as String,
-      createdAt: DateTime.parse(json['createdAt'] as String),
-      isCompleted: json['isCompleted'] as bool? ?? false,
-    );
-  }}
-
-  /// Create a copy with updated fields
   Item copyWith({{
     String? id,
     String? title,
     String? description,
     DateTime? createdAt,
     bool? isCompleted,
-  }}) {{
-    return Item(
-      id: id ?? this.id,
-      title: title ?? this.title,
-      description: description ?? this.description,
-      createdAt: createdAt ?? this.createdAt,
-      isCompleted: isCompleted ?? this.isCompleted,
-    );
-  }}
+  }}) => Item(
+    id: id ?? this.id,
+    title: title ?? this.title,
+    description: description ?? this.description,
+    createdAt: createdAt ?? this.createdAt,
+    isCompleted: isCompleted ?? this.isCompleted,
+  );
 }}
 
 /// User model
-/// Represents a user in the application
 class User {{
   final String id;
   final String name;
   final String email;
-  final String? avatarUrl;
 
   const User({{
     required this.id,
     required this.name,
     required this.email,
-    this.avatarUrl,
   }});
 
-  /// Convert user to JSON
-  Map<String, dynamic> toJson() {{
-    return {{
-      'id': id,
-      'name': name,
-      'email': email,
-      'avatarUrl': avatarUrl,
-    }};
-  }}
-
-  /// Create user from JSON
-  factory User.fromJson(Map<String, dynamic> json) {{
-    return User(
-      id: json['id'] as String,
-      name: json['name'] as String,
-      email: json['email'] as String,
-      avatarUrl: json['avatarUrl'] as String?,
-    );
-  }}
+  Map<String, dynamic> toJson() => {{
+    'id': id,
+    'name': name,
+    'email': email,
+  }};
+
+  factory User.fromJson(Map<String, dynamic> json) => User(
+    id: json['id'],
+    name: json['name'],
+    email: json['email'],
+  );
 }}
-"""
+```"""
 
+    ai_generated_code = generate_code_with_ai(prompt, max_length=800)
 
-def generate_services_dart(description):
-    """Generate the services.dart file content."""
-    return f"""// Business Logic and Services
+    # Add header comment
+    header = f"""// Data Models for the Flutter Application
 // Description: {description}
 // Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+// Note: This is AI-generated code - review and test before production use
 
+"""
+
+    return header + ai_generated_code
+
+
+def generate_services_dart(description):
+    """Generate the services.dart file content using AI."""
+    prompt = f"""Create Flutter service classes for an app described as: {description}
+
+Create service classes with:
+- Business logic methods
+- Data management operations (CRUD)
+- Validation utilities
+- Proper error handling
+- Based on the app requirements from the description
+
+```dart
 import 'models.dart';
 
-/// Data service for managing items
-/// This service handles CRUD operations for items
-class ItemService {{
+/// Data service for managing application data
+class DataService {{
   final List<Item> _items = [];
 
   /// Get all items
-  List<Item> getAllItems() {{
-    return List.unmodifiable(_items);
-  }}
+  List<Item> getAllItems() => List.unmodifiable(_items);
 
-  /// Add a new item
-  void addItem(Item item) {{
-    _items.add(item);
-  }}
+  /// Add new item
+  void addItem(Item item) => _items.add(item);
 
-  /// Update an existing item
+  /// Update existing item
   bool updateItem(String id, Item updatedItem) {{
     final index = _items.indexWhere((item) => item.id == id);
     if (index != -1) {{
@@ -460,7 +438,7 @@ class ItemService {{
     return false;
   }}
 
-  /// Delete an item
+  /// Delete item
   bool deleteItem(String id) {{
     final index = _items.indexWhere((item) => item.id == id);
     if (index != -1) {{
@@ -470,8 +448,8 @@ class ItemService {{
     return false;
   }}
 
-  /// Toggle item completion status
-  bool toggleItemCompletion(String id) {{
+  /// Toggle completion status
+  bool toggleCompletion(String id) {{
     final index = _items.indexWhere((item) => item.id == id);
     if (index != -1) {{
       _items[index].isCompleted = !_items[index].isCompleted;
@@ -479,41 +457,36 @@ class ItemService {{
     }}
     return false;
   }}
-
-  /// Get completed items
-  List<Item> getCompletedItems() {{
-    return _items.where((item) => item.isCompleted).toList();
-  }}
-
-  /// Get pending items
-  List<Item> getPendingItems() {{
-    return _items.where((item) => !item.isCompleted).toList();
-  }}
 }}
 
-/// Validation service
-/// Provides validation utilities
+/// Validation service with utility methods
 class ValidationService {{
-  /// Validate email format
   static bool isValidEmail(String email) {{
-    final emailRegex = RegExp(
-      r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{{2,}}$',
-    );
-    return emailRegex.hasMatch(email);
+    return RegExp(r'^[^@]+@[^@]+\\.[^@]+').hasMatch(email);
   }}
 
-  /// Validate string is not empty
   static bool isNotEmpty(String? value) {{
     return value != null && value.trim().isNotEmpty;
   }}
 
-  /// Validate minimum length
   static bool hasMinLength(String value, int minLength) {{
     return value.length >= minLength;
   }}
 }}
+```"""
+
+    ai_generated_code = generate_code_with_ai(prompt, max_length=800)
+
+    # Add header comment
+    header = f"""// Business Logic and Services
+// Description: {description}
+// Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+// Note: This is AI-generated code - review and test before production use
+
 """
 
+    return header + ai_generated_code
+
 
 def generate_pubspec(description):
     """Generate the pubspec.yaml file content."""
@@ -748,10 +721,11 @@ def create_interface():
             - **README.md** - Setup and usage instructions
             
             ### ⚠️ Important Notes:
-            - This is a **placeholder implementation** for demonstration
-            - In production, this would integrate with an AI model API (Claude, GPT, etc.)
+            - This uses a **small AI model** (DistilGPT-2) running directly on Hugging Face Spaces
+            - The AI generates Flutter/Dart code based on your description
             - Always review and test generated code before production use
-            - Customize the code to match your specific requirements
+            - The model may produce varying quality results - iterate on your descriptions
+            - Customize the generated code to match your specific requirements
             
             ### 🔧 Next Steps After Download:
             1. Extract the ZIP file