awacke1 commited on
Commit
d4fbf0d
·
verified ·
1 Parent(s): 29b8d1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- result = f"### 🔎 {q.replace('\n', ' ')}\n\n{r2}\n\n{refs}"
 
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: {q.replace('\n', ' ')}. {clean_for_speech(r2)} {clean_for_speech(refs)}"
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(q.replace('\n', ' '), result, "md")
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