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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|