Update app.py

app.py

@@ -10,7 +10,7 @@ st.set_page_config(page_title="JIGPC Chat Bot")
 # Hugging Face Credentials
 with st.sidebar:
     st.title('JIGPC Chat Bot')
-    st.write('This is a generative AI Chat Bot.')
+    #st.write('This is a generative AI Chat Bot.')
     
     # Use Hugging Face API Key from secrets or environment
     api_key = os.getenv("llama3")
@@ -29,19 +29,30 @@ with st.sidebar:
 
     st.subheader('Models and parameters')
     
-    # Model selection
-    model_options = [
-        "meta-llama/Llama-3.2-1B", 
-        "meta-llama/Llama-3.2-1B-Instruct", 
-        "meta-llama/Llama-3.2-3B-Instruct", 
-        "meta-llama/Llama-3.2-3B"
-    ]
-    
-    selected_model = st.sidebar.selectbox('Choose a Llama model', model_options, key='selected_model')
+    # Model selection categories
+    model_options = {
+        "Basic": [
+            "meta-llama/Llama-3.2-1B"
+        ],
+        "Basic-Medium": [
+            "meta-llama/Llama-3.2-1B-Instruct"
+        ],
+        "Medium-Fine": [
+            "meta-llama/Llama-3.2-3B-Instruct"
+        ],
+        "Finest": [
+            "meta-llama/Llama-3.2-3B"
+        ]
+    }
+
+    # Select category and model
+    selected_category = st.sidebar.selectbox('Select Model Category', ["Basic", "Basic-Medium", "Medium-Fine", "Finest"], index=0)
+    selected_model = st.sidebar.selectbox('Choose a Llama model', model_options[selected_category], key='selected_model')
 
+    # Slider inputs for parameters
     temperature = st.sidebar.slider('temperature', min_value=0.01, max_value=1.0, value=0.3, step=0.01)
     top_p = st.sidebar.slider('top_p', min_value=0.01, max_value=1.0, value=0.9, step=0.01)
-    max_length = st.sidebar.slider('max_length', min_value=20, max_value=80, value=100, step=5)
+    max_length = st.sidebar.slider('max_length', min_value=20, max_value=100, value=60, step=5)
     
 
 # Store LLM generated responses