Update app.py

app.py

@@ -243,7 +243,7 @@ def search_glossary(query):
             "mistralai/Mixtral-8x7B-Instruct-v0.1",	# LLM Model Dropdown component
             api_name="/update_with_rag_md"
         )
-        st.code(response1[0], language="python", *, line_numbers=True, wrap_lines=False)
+        st.code(response1[0], language="python", line_numbers=True, wrap_lines=False)
 
         
         # ArXiv searcher - Paper Summary & Ask LLM
@@ -256,13 +256,15 @@ def search_glossary(query):
             llm_model_picked=model_choice,
             api_name="/update_with_rag_md"
         )
-        
+        st.code(response1, language="python", line_numbers=True, wrap_lines=False)
+
         response2 = client.predict(
             prompt=query,
             llm_model_picked=model_choice,
             stream_outputs=True,
             api_name="/ask_llm"
         )
+        st.code(response2, language="python", line_numbers=True, wrap_lines=False)
 
         # Aggregate hyperlinks and show with emojis
         hyperlinks = extract_hyperlinks([response1, response2])