fix latex

app.py

@@ -6,7 +6,6 @@ import streamlit as st
 
 # from openai import OpenAI
 import random
-# import prompts
 
 from utils.help import get_disclaimer
 from utils.format import sec_to_time, fix_latex, get_youtube_embed
@@ -171,9 +170,6 @@ text_data_YT, context_embeddings_YT = load_youtube_data(base_path, model_name, y
 text_data_Latex, context_embeddings_Latex = load_book_data(base_path, model_name, latex_chunk_tokens, latex_overlap_tokens)
 summary = load_summary('data/KG_FEM_summary.json')
 
-
-
-
 if 'question_answered' not in st.session_state:
     st.session_state.question_answered = False
 if 'context_by_video' not in st.session_state:
@@ -268,6 +264,7 @@ if submit_button_placeholder.button("AI Answer", type="primary"):
                 temperature=integration_temperature,
                 top_p=integration_top_p
             )
+            answer = fix_latex(answer) 
 
         if answer.split()[0] == "NOT_ENOUGH_INFO":
             st.markdown("")
@@ -297,7 +294,7 @@ if submit_button_placeholder.button("AI Answer", type="primary"):
 if st.session_state.question_answered:
     st.markdown("")
     st.markdown("#### Query:")
-    st.markdown(prompts.fix_latex(st.session_state.question))
+    st.markdown(fix_latex(st.session_state.question))
     if show_expert_responce:
         st.markdown("#### Initial Expert Answer:")
         st.markdown(st.session_state.expert_answer)

utils/openai_utils.py

@@ -41,7 +41,7 @@ def openai_domain_specific_answer_generation(system_prompt, question, model="gpt
         presence_penalty=0.0          # Prevent introduction of unrelated ideas.
     )
     
-    return fix_latex(response.choices[0].message.content)
+    return response.choices[0].message.content
 
 def openai_context_integration(system_prompt, query, expert_answer, retrieved_context, model="gpt4o-mini", temperature=0.3, top_p=0.3):
     
@@ -76,4 +76,4 @@ def openai_context_integration(system_prompt, query, expert_answer, retrieved_co
         presence_penalty=0.0          # Neutral to avoid introducing unrelated ideas.
     )
     
-    return fix_latex(response.choices[0].message.content)
+    return response.choices[0].message.content