-
Notifications
You must be signed in to change notification settings - Fork 15
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #4 from varunon9/content_generator
New feature: Generate content e.g. poem, quotes, thoughts etc and share with world.
- Loading branch information
Showing
17 changed files
with
493 additions
and
34 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -45,3 +45,6 @@ app.*.map.json | |
| /android/app/debug | ||
| /android/app/profile | ||
| /android/app/release | ||
|
|
||
| # Scripts | ||
| /server | ||
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
237 changes: 237 additions & 0 deletions
237
lib/src/modules/content_generator/screens/content_generator_screen.dart
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,237 @@ | ||
| import 'package:bubble/bubble.dart'; | ||
| import 'package:cloud_firestore/cloud_firestore.dart'; | ||
| import 'package:flutter/material.dart'; | ||
| import 'package:flutter/services.dart'; | ||
| import 'package:flutter_markdown/flutter_markdown.dart'; | ||
| import 'package:pocket_ai/src/constants.dart'; | ||
| import 'package:pocket_ai/src/globals.dart'; | ||
| import 'package:pocket_ai/src/modules/chat/chat_actions.dart'; | ||
| import 'package:pocket_ai/src/modules/chat/models/chat_message.dart'; | ||
| import 'package:pocket_ai/src/modules/content_generator/widgets/content_image.dart'; | ||
| import 'package:pocket_ai/src/modules/faqs/screens/faqs_screen.dart'; | ||
| import 'package:pocket_ai/src/modules/settings/screens/settings_screen.dart'; | ||
| import 'package:pocket_ai/src/utils/analytics.dart'; | ||
| import 'package:pocket_ai/src/utils/common.dart'; | ||
| import 'package:pocket_ai/src/widgets/custom_colors.dart'; | ||
| import 'package:pocket_ai/src/widgets/custom_text.dart'; | ||
| import 'package:pocket_ai/src/widgets/custom_text_form_field.dart'; | ||
| import 'package:pocket_ai/src/widgets/heading.dart'; | ||
|
|
||
| class ContentGeneratorScreen extends StatefulWidget { | ||
| static const routeName = '/content-generator'; | ||
|
|
||
| const ContentGeneratorScreen({Key? key}) : super(key: key); | ||
|
|
||
| @override | ||
| State<StatefulWidget> createState() => _ContentGeneratorScreen(); | ||
| } | ||
|
|
||
| class _ContentGeneratorScreen extends State<ContentGeneratorScreen> { | ||
| List<ChatMessage> chatMessages = [ | ||
| ChatMessage( | ||
| content: AiBotConstants.introMessageForContentGenerator, | ||
| role: ChatRole.system), | ||
| ChatMessage( | ||
| content: AiBotConstants.contentGeneratorSamplePrompt, | ||
| role: ChatRole.system) | ||
| ]; | ||
| bool apiCallInProgress = false; | ||
| FirebaseFirestore db = FirebaseFirestore.instance; | ||
|
|
||
| TextEditingController contentGeneratorPromptController = | ||
| TextEditingController(); | ||
| ScrollController listViewController = ScrollController(); | ||
|
|
||
| @override | ||
| void initState() { | ||
| super.initState(); | ||
| logEvent(EventNames.contentGeneratorScreenViewed, {}); | ||
| } | ||
|
|
||
| void onChatMessageLongPress(ChatMessage chatItem) { | ||
| Clipboard.setData(ClipboardData(text: chatItem.content)).then((value) { | ||
| showToastMessage('Copied to Clipboard'); | ||
| }); | ||
| } | ||
|
|
||
| void saveContentGeneratorPromptsToFirestore(String prompt) { | ||
| // store prompts to Firestore for study & analytics | ||
| String? deviceId = Globals.deviceId; | ||
| if (deviceId != null) { | ||
| db | ||
| .collection(FirestoreCollectionsConst.contentGeneratorPrompts) | ||
| .doc(deviceId) | ||
| .collection(FirestoreCollectionsConst.prompts) | ||
| .doc() | ||
| .set({'prompt': prompt, 'time': FieldValue.serverTimestamp()}); | ||
| } | ||
| } | ||
|
|
||
| void onSendPress() { | ||
| String prompt = contentGeneratorPromptController.text; | ||
| if (prompt.isEmpty) { | ||
| return; | ||
| } | ||
| logEvent(EventNames.generateContentClicked, {}); | ||
|
|
||
| setState(() { | ||
| chatMessages = [ | ||
| ...chatMessages, | ||
| ChatMessage(content: prompt, role: ChatRole.user) | ||
| ]; | ||
| apiCallInProgress = true; | ||
| }); | ||
| contentGeneratorPromptController.text = ''; | ||
|
|
||
| // adding delay so that list view is scrolled after setState re-render has been completed | ||
| Future.delayed(const Duration(milliseconds: 100), () { | ||
| listViewController.jumpTo(listViewController.position.maxScrollExtent); | ||
| }); | ||
| saveContentGeneratorPromptsToFirestore(prompt); | ||
|
|
||
| getResponseFromOpenAi([ChatMessage(content: prompt, role: ChatRole.user)]) | ||
| .then((response) { | ||
| String botMessage = '${response['choices'][0]['message']['content']}'; | ||
| setState(() { | ||
| chatMessages = [ | ||
| ...chatMessages, | ||
| ChatMessage(content: botMessage, role: ChatRole.assistant) | ||
| ]; | ||
| }); | ||
| logEvent(EventNames.openAiResponseSuccess, {}); | ||
| }).catchError((error) { | ||
| logApiErrorAndShowMessage(context, exception: error); | ||
| logEvent(EventNames.openAiResponseFailed, {}); | ||
| }).then((value) { | ||
| setState(() { | ||
| apiCallInProgress = false; | ||
| }); | ||
| Future.delayed(const Duration(milliseconds: 100), () { | ||
| listViewController.jumpTo(listViewController.position.maxScrollExtent); | ||
| }); | ||
| }); | ||
| } | ||
|
|
||
| void handlePopupMenuClick(int item) { | ||
| switch (item) { | ||
| case 0: | ||
| logEvent(EventNames.helpIconClicked, {}); | ||
| navigateToScreen(context, FaqsScreen.routeName); | ||
| break; | ||
| case 1: | ||
| logEvent(EventNames.settingsIconClicked, {}); | ||
| navigateToScreen(context, SettingsScreen.routeName); | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| @override | ||
| Widget build(BuildContext context) { | ||
| return Scaffold( | ||
| appBar: AppBar( | ||
| automaticallyImplyLeading: false, | ||
| title: const Heading( | ||
| 'Content Generator', | ||
| type: HeadingType.h4, | ||
| ), | ||
| backgroundColor: CustomColors.darkBackground, | ||
| actions: <Widget>[ | ||
| PopupMenuButton<int>( | ||
| onSelected: (item) => handlePopupMenuClick(item), | ||
| itemBuilder: (context) => [ | ||
| const PopupMenuItem<int>(value: 0, child: Text('Help')), | ||
| const PopupMenuItem<int>(value: 1, child: Text('Settings')), | ||
| ], | ||
| ), | ||
| ]), | ||
| body: Stack(children: [ | ||
| Container( | ||
| margin: const EdgeInsets.only(bottom: 72), | ||
| child: ListView.builder( | ||
| controller: listViewController, | ||
| itemCount: chatMessages.length, | ||
| shrinkWrap: true, | ||
| physics: const AlwaysScrollableScrollPhysics(), | ||
| padding: const EdgeInsets.only(top: 12, bottom: 12), | ||
| itemBuilder: (context, index) { | ||
| var chatItem = chatMessages[index]; | ||
| bool fromBot = chatItem.role == ChatRole.assistant; | ||
| bool fromSystem = chatItem.role == ChatRole.system; | ||
| return GestureDetector( | ||
| onLongPress: () { | ||
| onChatMessageLongPress(chatItem); | ||
| }, | ||
| child: fromSystem | ||
| ? Container( | ||
| margin: const EdgeInsets.only( | ||
| top: 8, bottom: 0, left: 16, right: 16), | ||
| padding: const EdgeInsets.all(8), | ||
| decoration: BoxDecoration( | ||
| borderRadius: BorderRadius.circular(8), | ||
| color: CustomColors.secondary, | ||
| ), | ||
| child: CustomText( | ||
| chatItem.content, | ||
| style: const TextStyle( | ||
| color: Colors.white, fontSize: 10), | ||
| ), | ||
| ) | ||
| : Bubble( | ||
| nip: fromBot | ||
| ? BubbleNip.leftTop | ||
| : BubbleNip.rightTop, | ||
| margin: const BubbleEdges.only( | ||
| top: 16, left: 8, right: 16), | ||
| color: | ||
| fromBot ? Colors.white : CustomColors.lightText, | ||
| alignment: fromBot | ||
| ? Alignment.topLeft | ||
| : Alignment.topRight, | ||
| child: fromBot | ||
| ? ContentImage(content: chatItem.content) | ||
| : MarkdownBody(data: chatItem.content)), | ||
| ); | ||
| })), | ||
| Align( | ||
| alignment: Alignment.bottomLeft, | ||
| child: Container( | ||
| margin: | ||
| const EdgeInsets.only(top: 8, left: 8, right: 8, bottom: 8), | ||
| child: Row( | ||
| children: [ | ||
| Expanded( | ||
| child: Container( | ||
| margin: const EdgeInsets.only(right: 8), | ||
| child: CustomTextFormField( | ||
| onChanged: (value) => {}, | ||
| controller: contentGeneratorPromptController, | ||
| minLines: 1, | ||
| maxLines: 4, | ||
| textInputType: TextInputType.multiline, | ||
| hintText: 'Your prompt to generate content'), | ||
| ), | ||
| ), | ||
| Ink( | ||
| decoration: const ShapeDecoration( | ||
| color: CustomColors.secondary, | ||
| shape: CircleBorder(), | ||
| ), | ||
| width: 48, | ||
| height: 48, | ||
| child: apiCallInProgress | ||
| ? const CircularProgressIndicator( | ||
| color: CustomColors.primary, | ||
| ) | ||
| : IconButton( | ||
| tooltip: 'Send', | ||
| onPressed: onSendPress, | ||
| color: CustomColors.primary, | ||
| icon: const Icon(Icons.send_rounded)), | ||
| ) | ||
| ], | ||
| )), | ||
| ) | ||
| ]), | ||
| ); | ||
| } | ||
| } |
Oops, something went wrong.