Skip to content

Commit

Permalink
start on huggingface dialog
Browse files Browse the repository at this point in the history
  • Loading branch information
danemadsen committed Jun 25, 2024
1 parent bc735b7 commit 221f9c3
Show file tree
Hide file tree
Showing 2 changed files with 52 additions and 11 deletions.
50 changes: 50 additions & 0 deletions lib/ui/desktop/buttons/huggingface_button.dart
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
import 'package:flutter/material.dart';
import 'package:flutter_svg/svg.dart';

class HuggingfaceButton extends StatefulWidget {
const HuggingfaceButton({super.key});

@override
State<HuggingfaceButton> createState() => _HuggingfaceButtonState();
}

class _HuggingfaceButtonState extends State<HuggingfaceButton> {
@override
Widget build(BuildContext context) {
return IconButton(
tooltip: 'HuggingFace',
onPressed: () => showHuggingfaceDialog(context),
icon: SizedBox(
width: 30.0,
height: 30.0,
child: SvgPicture.asset(
'assets/huggingface-colour.svg',
),
)
);
}

void showHuggingfaceDialog(BuildContext context) {
showDialog(
context: context,
builder: (BuildContext context) {
return AlertDialog(
title: const Text(
'Select HuggingFace Model',
textAlign: TextAlign.center
),
content: const Text('This is a HuggingFace button.'),
actions: [
FilledButton(
onPressed: () => Navigator.of(context).pop(),
child: const Text(
"Close"
),
),
],
actionsAlignment: MainAxisAlignment.center,
);
},
);
}
}
13 changes: 2 additions & 11 deletions lib/ui/desktop/layout/home_app_bar.dart
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ import 'package:flutter/material.dart';
import 'package:maid/classes/providers/large_language_models/llama_cpp_model.dart';
import 'package:maid/enumerators/large_language_model_type.dart';
import 'package:maid/classes/providers/app_data.dart';
import 'package:maid/ui/desktop/buttons/huggingface_button.dart';
import 'package:maid/ui/desktop/buttons/load_model_button.dart';
import 'package:maid/ui/desktop/dropdowns/model_dropdown.dart';
import 'package:maid/ui/shared/dropdowns/llm_platform_dropdown.dart';
Expand Down Expand Up @@ -55,17 +56,7 @@ class HomeAppBar extends StatelessWidget implements PreferredSizeWidget {

List<Widget> llamaCppWidgets(BuildContext context) {
return [
IconButton(
tooltip: 'HuggingFace',
onPressed: () {},
icon: SizedBox(
width: 30.0,
height: 30.0,
child: SvgPicture.asset(
'assets/huggingface-colour.svg',
),
)
),
const HuggingfaceButton(),
const Expanded(
flex: 8,
child: LoadModelButton(),
Expand Down

0 comments on commit 221f9c3

Please sign in to comment.