Spaces:
Running
Running
Remove old app file
Browse files- legacy_app.py +0 -294
legacy_app.py
DELETED
|
@@ -1,294 +0,0 @@
|
|
| 1 |
-
import pathlib
|
| 2 |
-
import logging
|
| 3 |
-
import tempfile
|
| 4 |
-
from typing import List, Tuple
|
| 5 |
-
|
| 6 |
-
import json5
|
| 7 |
-
import metaphor_python as metaphor
|
| 8 |
-
import streamlit as st
|
| 9 |
-
|
| 10 |
-
from helpers import llm_helper, pptx_helper
|
| 11 |
-
from global_config import GlobalConfig
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
APP_TEXT = json5.loads(open(GlobalConfig.APP_STRINGS_FILE, 'r', encoding='utf-8').read())
|
| 15 |
-
GB_CONVERTER = 2 ** 30
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
logger = logging.getLogger(__name__)
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
@st.cache_data
|
| 22 |
-
def get_contents_wrapper(text: str) -> str:
|
| 23 |
-
"""
|
| 24 |
-
Fetch and cache the slide deck contents on a topic by calling an external API.
|
| 25 |
-
|
| 26 |
-
:param text: The presentation topic.
|
| 27 |
-
:return: The slide deck contents or outline in JSON format.
|
| 28 |
-
"""
|
| 29 |
-
|
| 30 |
-
logger.info('LLM call because of cache miss...')
|
| 31 |
-
return llm_helper.generate_slides_content(text).strip()
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
@st.cache_resource
|
| 35 |
-
def get_metaphor_client_wrapper() -> metaphor.Metaphor:
|
| 36 |
-
"""
|
| 37 |
-
Create a Metaphor client for semantic Web search.
|
| 38 |
-
|
| 39 |
-
:return: Metaphor instance.
|
| 40 |
-
"""
|
| 41 |
-
|
| 42 |
-
return metaphor.Metaphor(api_key=GlobalConfig.METAPHOR_API_KEY)
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
@st.cache_data
|
| 46 |
-
def get_web_search_results_wrapper(text: str) -> List[Tuple[str, str]]:
|
| 47 |
-
"""
|
| 48 |
-
Fetch and cache the Web search results on a given topic.
|
| 49 |
-
|
| 50 |
-
:param text: The topic.
|
| 51 |
-
:return: A list of (title, link) tuples.
|
| 52 |
-
"""
|
| 53 |
-
|
| 54 |
-
results = []
|
| 55 |
-
search_results = get_metaphor_client_wrapper().search(
|
| 56 |
-
text,
|
| 57 |
-
use_autoprompt=True,
|
| 58 |
-
num_results=5
|
| 59 |
-
)
|
| 60 |
-
|
| 61 |
-
for a_result in search_results.results:
|
| 62 |
-
results.append((a_result.title, a_result.url))
|
| 63 |
-
|
| 64 |
-
return results
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
def build_ui():
|
| 68 |
-
"""
|
| 69 |
-
Display the input elements for content generation. Only covers the first step.
|
| 70 |
-
"""
|
| 71 |
-
|
| 72 |
-
# get_disk_used_percentage()
|
| 73 |
-
|
| 74 |
-
st.title(APP_TEXT['app_name'])
|
| 75 |
-
st.subheader(APP_TEXT['caption'])
|
| 76 |
-
st.markdown(
|
| 77 |
-
'Powered by'
|
| 78 |
-
' [Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2).'
|
| 79 |
-
)
|
| 80 |
-
st.markdown(
|
| 81 |
-
'*If the JSON is generated or parsed incorrectly, try again later by making minor changes'
|
| 82 |
-
' to the input text.*'
|
| 83 |
-
)
|
| 84 |
-
|
| 85 |
-
with st.form('my_form'):
|
| 86 |
-
# Topic input
|
| 87 |
-
try:
|
| 88 |
-
with open(GlobalConfig.PRELOAD_DATA_FILE, 'r', encoding='utf-8') as in_file:
|
| 89 |
-
preload_data = json5.loads(in_file.read())
|
| 90 |
-
except (FileExistsError, FileNotFoundError):
|
| 91 |
-
preload_data = {'topic': '', 'audience': ''}
|
| 92 |
-
|
| 93 |
-
topic = st.text_area(
|
| 94 |
-
APP_TEXT['input_labels'][0],
|
| 95 |
-
value=preload_data['topic']
|
| 96 |
-
)
|
| 97 |
-
|
| 98 |
-
texts = list(GlobalConfig.PPTX_TEMPLATE_FILES.keys())
|
| 99 |
-
captions = [GlobalConfig.PPTX_TEMPLATE_FILES[x]['caption'] for x in texts]
|
| 100 |
-
|
| 101 |
-
pptx_template = st.radio(
|
| 102 |
-
'Select a presentation template:',
|
| 103 |
-
texts,
|
| 104 |
-
captions=captions,
|
| 105 |
-
horizontal=True
|
| 106 |
-
)
|
| 107 |
-
|
| 108 |
-
st.divider()
|
| 109 |
-
submit = st.form_submit_button('Generate slide deck')
|
| 110 |
-
|
| 111 |
-
if submit:
|
| 112 |
-
# st.write(f'Clicked {time.time()}')
|
| 113 |
-
st.session_state.submitted = True
|
| 114 |
-
|
| 115 |
-
# https://github.com/streamlit/streamlit/issues/3832#issuecomment-1138994421
|
| 116 |
-
if 'submitted' in st.session_state:
|
| 117 |
-
progress_text = 'Generating the slides...give it a moment'
|
| 118 |
-
progress_bar = st.progress(0, text=progress_text)
|
| 119 |
-
|
| 120 |
-
topic_txt = topic.strip()
|
| 121 |
-
generate_presentation(topic_txt, pptx_template, progress_bar)
|
| 122 |
-
|
| 123 |
-
st.divider()
|
| 124 |
-
st.text(APP_TEXT['tos'])
|
| 125 |
-
st.text(APP_TEXT['tos2'])
|
| 126 |
-
|
| 127 |
-
st.markdown(
|
| 128 |
-
'![Visitors]'
|
| 129 |
-
'(https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fbarunsaha%2Fslide-deck-ai&countColor=%23263759)'
|
| 130 |
-
)
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
def generate_presentation(topic: str, pptx_template: str, progress_bar):
|
| 134 |
-
"""
|
| 135 |
-
Process the inputs to generate the slides.
|
| 136 |
-
|
| 137 |
-
:param topic: The presentation topic based on which contents are to be generated.
|
| 138 |
-
:param pptx_template: The PowerPoint template name to be used.
|
| 139 |
-
:param progress_bar: Progress bar from the page.
|
| 140 |
-
"""
|
| 141 |
-
|
| 142 |
-
topic_length = len(topic)
|
| 143 |
-
logger.debug('Input length:: topic: %s', topic_length)
|
| 144 |
-
|
| 145 |
-
if topic_length >= 10:
|
| 146 |
-
logger.debug('Topic: %s', topic)
|
| 147 |
-
target_length = min(topic_length, GlobalConfig.LLM_MODEL_MAX_INPUT_LENGTH)
|
| 148 |
-
|
| 149 |
-
try:
|
| 150 |
-
# Step 1: Generate the contents in JSON format using an LLM
|
| 151 |
-
json_str = process_slides_contents(topic[:target_length], progress_bar)
|
| 152 |
-
logger.debug('Truncated topic: %s', topic[:target_length])
|
| 153 |
-
logger.debug('Length of JSON: %d', len(json_str))
|
| 154 |
-
|
| 155 |
-
# Step 2: Generate the slide deck based on the template specified
|
| 156 |
-
if len(json_str) > 0:
|
| 157 |
-
st.info(
|
| 158 |
-
'Tip: The generated content doesn\'t look so great?'
|
| 159 |
-
' Need alternatives? Just change your description text and try again.',
|
| 160 |
-
icon="💡️"
|
| 161 |
-
)
|
| 162 |
-
else:
|
| 163 |
-
st.error(
|
| 164 |
-
'Unfortunately, JSON generation failed, so the next steps would lead'
|
| 165 |
-
' to nowhere. Try again or come back later.'
|
| 166 |
-
)
|
| 167 |
-
return
|
| 168 |
-
|
| 169 |
-
all_headers = generate_slide_deck(json_str, pptx_template, progress_bar)
|
| 170 |
-
|
| 171 |
-
# Step 3: Bonus stuff: Web references and AI art
|
| 172 |
-
show_bonus_stuff(all_headers)
|
| 173 |
-
|
| 174 |
-
except ValueError as ve:
|
| 175 |
-
st.error(f'Unfortunately, an error occurred: {ve}! '
|
| 176 |
-
f'Please change the text, try again later, or report it, sharing your inputs.')
|
| 177 |
-
|
| 178 |
-
else:
|
| 179 |
-
st.error('Not enough information provided! Please be little more descriptive :)')
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
def process_slides_contents(text: str, progress_bar: st.progress) -> str:
|
| 183 |
-
"""
|
| 184 |
-
Convert given text into structured data and display. Update the UI.
|
| 185 |
-
|
| 186 |
-
:param text: The topic description for the presentation.
|
| 187 |
-
:param progress_bar: Progress bar for this step.
|
| 188 |
-
:return: The contents as a JSON-formatted string.
|
| 189 |
-
"""
|
| 190 |
-
|
| 191 |
-
json_str = ''
|
| 192 |
-
|
| 193 |
-
try:
|
| 194 |
-
logger.info('Calling LLM for content generation on the topic: %s', text)
|
| 195 |
-
json_str = get_contents_wrapper(text)
|
| 196 |
-
except Exception as ex:
|
| 197 |
-
st.error(
|
| 198 |
-
f'An exception occurred while trying to convert to JSON. It could be because of heavy'
|
| 199 |
-
f' traffic or something else. Try doing it again or try again later.'
|
| 200 |
-
f'\nError message: {ex}'
|
| 201 |
-
)
|
| 202 |
-
|
| 203 |
-
progress_bar.progress(50, text='Contents generated')
|
| 204 |
-
|
| 205 |
-
with st.expander('The generated contents (in JSON format)'):
|
| 206 |
-
st.code(json_str, language='json')
|
| 207 |
-
|
| 208 |
-
return json_str
|
| 209 |
-
|
| 210 |
-
|
| 211 |
-
def generate_slide_deck(json_str: str, pptx_template: str, progress_bar) -> List:
|
| 212 |
-
"""
|
| 213 |
-
Create a slide deck.
|
| 214 |
-
|
| 215 |
-
:param json_str: The contents in JSON format.
|
| 216 |
-
:param pptx_template: The PPTX template name.
|
| 217 |
-
:param progress_bar: Progress bar.
|
| 218 |
-
:return: A list of all slide headers and the title.
|
| 219 |
-
"""
|
| 220 |
-
|
| 221 |
-
progress_text = 'Creating the slide deck...give it a moment'
|
| 222 |
-
progress_bar.progress(75, text=progress_text)
|
| 223 |
-
|
| 224 |
-
# # Get a unique name for the file to save -- use the session ID
|
| 225 |
-
# ctx = st_sr.get_script_run_ctx()
|
| 226 |
-
# session_id = ctx.session_id
|
| 227 |
-
# timestamp = time.time()
|
| 228 |
-
# output_file_name = f'{session_id}_{timestamp}.pptx'
|
| 229 |
-
|
| 230 |
-
temp = tempfile.NamedTemporaryFile(delete=False, suffix='.pptx')
|
| 231 |
-
path = pathlib.Path(temp.name)
|
| 232 |
-
|
| 233 |
-
logger.info('Creating PPTX file...')
|
| 234 |
-
all_headers = pptx_helper.generate_powerpoint_presentation(
|
| 235 |
-
json_str,
|
| 236 |
-
slides_template=pptx_template,
|
| 237 |
-
output_file_path=path
|
| 238 |
-
)
|
| 239 |
-
progress_bar.progress(100, text='Done!')
|
| 240 |
-
|
| 241 |
-
with open(path, 'rb') as f:
|
| 242 |
-
st.download_button('Download PPTX file', f, file_name='Presentation.pptx')
|
| 243 |
-
|
| 244 |
-
if temp:
|
| 245 |
-
temp.close()
|
| 246 |
-
|
| 247 |
-
return all_headers
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
def show_bonus_stuff(ppt_headers: List[str]):
|
| 251 |
-
"""
|
| 252 |
-
Show bonus stuff for the presentation.
|
| 253 |
-
|
| 254 |
-
:param ppt_headers: A list of the slide headings.
|
| 255 |
-
"""
|
| 256 |
-
|
| 257 |
-
# Use the presentation title and the slide headers to find relevant info online
|
| 258 |
-
logger.info('Calling Metaphor search...')
|
| 259 |
-
ppt_text = ' '.join(ppt_headers)
|
| 260 |
-
search_results = get_web_search_results_wrapper(ppt_text)
|
| 261 |
-
md_text_items = []
|
| 262 |
-
|
| 263 |
-
for (title, link) in search_results:
|
| 264 |
-
md_text_items.append(f'[{title}]({link})')
|
| 265 |
-
|
| 266 |
-
with st.expander('Related Web references'):
|
| 267 |
-
st.markdown('\n\n'.join(md_text_items))
|
| 268 |
-
|
| 269 |
-
logger.info('Done!')
|
| 270 |
-
|
| 271 |
-
# # Avoid image generation. It costs time and an API call, so just limit to the text generation.
|
| 272 |
-
# with st.expander('AI-generated image on the presentation topic'):
|
| 273 |
-
# logger.info('Calling SDXL for image generation...')
|
| 274 |
-
# # img_empty.write('')
|
| 275 |
-
# # img_text.write(APP_TEXT['image_info'])
|
| 276 |
-
# image = get_ai_image_wrapper(ppt_text)
|
| 277 |
-
#
|
| 278 |
-
# if len(image) > 0:
|
| 279 |
-
# image = base64.b64decode(image)
|
| 280 |
-
# st.image(image, caption=ppt_text)
|
| 281 |
-
# st.info('Tip: Right-click on the image to save it.', icon="💡️")
|
| 282 |
-
# logger.info('Image added')
|
| 283 |
-
|
| 284 |
-
|
| 285 |
-
def main():
|
| 286 |
-
"""
|
| 287 |
-
Trigger application run.
|
| 288 |
-
"""
|
| 289 |
-
|
| 290 |
-
build_ui()
|
| 291 |
-
|
| 292 |
-
|
| 293 |
-
if __name__ == '__main__':
|
| 294 |
-
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|