Update app.py

app.py

@@ -66,6 +66,11 @@ with st.sidebar:
         "Insert a number (top n rows to be selected):", value=3, step=1
     )
 
+    # Select FM
+    option = st.selectbox(
+        "Which foundational model would you like?",
+        ("GPT4", "LLAMA3"))
+
     # Clear button
     clear_button = st.sidebar.button("Clear Conversation", key="clear")
 
@@ -129,8 +134,18 @@ elif uploaded_files:
             refs_tab = refs_tab.head(math.ceil(top_n))
             result = refs_tab
 
-            # Call GPT
-            response = call_gpt(prompt, ' '.join(list(result.sentences)))
+            # Call FM
+            content = ' '.join(list(result.sentences))
+            if option == "GPT4":
+                response = call_gpt(prompt, content)
+            else:
+                response = call_llama(
+                    f"""
+                    Answer the question: {prompt} 
+                    
+                    Use the following information: {content}
+                    """
+                )
 
             # Display assistant response in chat message container
             with st.chat_message("assistant"):