Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
@@ -353,6 +353,7 @@ def parse_arxiv_refs(ref_text: str):
|
|
353 |
break
|
354 |
return results
|
355 |
|
|
|
356 |
def perform_ai_lookup(q, vocal_summary=True, extended_refs=False,
|
357 |
titles_summary=True, full_audio=False, selected_voice="en-US-AriaNeural"):
|
358 |
"""Perform Arxiv search and generate audio summaries."""
|
@@ -364,12 +365,13 @@ def perform_ai_lookup(q, vocal_summary=True, extended_refs=False,
|
|
364 |
r2 = client.predict(q,"mistralai/Mixtral-8x7B-Instruct-v0.1",True,api_name="/ask_llm")
|
365 |
|
366 |
# 🎯 2) Combine for final text output
|
367 |
-
|
|
|
368 |
st.markdown(result)
|
369 |
|
370 |
# 🎯 3) Generate "all at once" audio if requested
|
371 |
if full_audio:
|
372 |
-
complete_text = f"Complete response for query: {
|
373 |
audio_file_full = speak_with_edge_tts(complete_text, selected_voice)
|
374 |
st.write("### 📚 Full Audio")
|
375 |
play_and_download_audio(audio_file_full)
|
@@ -432,7 +434,7 @@ def perform_ai_lookup(q, vocal_summary=True, extended_refs=False,
|
|
432 |
st.write(f"**Total Elapsed:** {elapsed:.2f} s")
|
433 |
|
434 |
# Always create a file with the result
|
435 |
-
create_file(
|
436 |
|
437 |
return result
|
438 |
|
|
|
353 |
break
|
354 |
return results
|
355 |
|
356 |
+
|
357 |
def perform_ai_lookup(q, vocal_summary=True, extended_refs=False,
|
358 |
titles_summary=True, full_audio=False, selected_voice="en-US-AriaNeural"):
|
359 |
"""Perform Arxiv search and generate audio summaries."""
|
|
|
365 |
r2 = client.predict(q,"mistralai/Mixtral-8x7B-Instruct-v0.1",True,api_name="/ask_llm")
|
366 |
|
367 |
# 🎯 2) Combine for final text output
|
368 |
+
clean_q = q.replace('\n', ' ')
|
369 |
+
result = f"### 🔎 {clean_q}\n\n{r2}\n\n{refs}"
|
370 |
st.markdown(result)
|
371 |
|
372 |
# 🎯 3) Generate "all at once" audio if requested
|
373 |
if full_audio:
|
374 |
+
complete_text = f"Complete response for query: {clean_q}. {clean_for_speech(r2)} {clean_for_speech(refs)}"
|
375 |
audio_file_full = speak_with_edge_tts(complete_text, selected_voice)
|
376 |
st.write("### 📚 Full Audio")
|
377 |
play_and_download_audio(audio_file_full)
|
|
|
434 |
st.write(f"**Total Elapsed:** {elapsed:.2f} s")
|
435 |
|
436 |
# Always create a file with the result
|
437 |
+
create_file(clean_q, result, "md")
|
438 |
|
439 |
return result
|
440 |
|