Upload app.py

app.py

@@ -12,12 +12,24 @@ def main():
 
     st.subheader("Intended Uses")
     st.write("This model is designed to assist publishers in understanding and exploring what modules most publishers use with their Prebid set-up. It can serve as a valuable reference to gain insights into common Prebid modules, best practices, and different approaches used by publishers across various domains. The model should be seen as a helpful tool to gain inspiration and understanding of common Prebid modules but not as a substitute for thorough testing and manual review of the final modules used.")
-    st.write("To learn more about the default model, visit the [Prebid_Module_GPT2 model page](https://huggingface.co/PeterBrendan/Prebid_Module_GPT2). You can also refer to the [official Prebid Documentation on pbjs.setConfig](https://docs.prebid.org/dev-docs/publisher-api-reference/setConfig.html) for more information.")
+    st.write("To learn more about the default model, visit the [Prebid_Module_GPT2 model page](https://huggingface.co/PeterBrendan/Prebid_Module_GPT2). You can also refer to the [official Prebid Documentation on modules](https://docs.prebid.org/dev-docs/modules/) for more information.")
 
     st.write("*Note:* The model may take some time to generate the output.")
 
-    # Create a text input field for user prompt
-    user_input = st.text_input("Enter a Prebid module:", "")
+    # Default Prebid modules
+    default_modules = ["[", "appnexusBidAdapter","ttdBidAdapter", "rubiconBidAdapter", "dfpAdServerVideo", "pubmaticBidAdapter", "gptPreAuction"]
+
+    # Create a selectbox for default prompts
+    default_module = st.selectbox("Choose a default Prebid module:", default_modules)
+
+    # Create a text input field for custom prompt
+    custom_module = st.text_input("Enter a custom Prebid module:", "")
+
+    # Check if a default module is selected
+    if default_module:
+        user_input = default_module
+    else:
+        user_input = custom_module
 
     # Check if the user input is empty
     if user_input: