add regions (#1)

- add regions (1f270b1d9ffcbcd5feb7bc2065030f827fe428d5)


Co-authored-by: Khadija Bayoud <[email protected]>

app.py

@@ -15,6 +15,23 @@ def get_text():
         input_text = st.text_input("You: ", "", key="input")
         return input_text
 
+soil_types = {
+    "Sais plain": "Brown limestone, vertisols, lithosols, and regosols",
+    "Chaouïa, Doukkala, and Abda plains": "Rendzines associated with lithosols in the Atlantic coast and isohumic and vertisols inland",
+    "Eastern High Plateaux and Moulouya Valley": "Sierozems and fluvisols",
+    "Rif": "Brown soils associated with lithosols and regosols or vertisols",
+    "Mamora and Zemmour plateau": "Sandy soil",
+    "Middle Atlas": "Brown soils and rendzinas",
+    "High Atlas": "Lithosols and regosols, in association with brown soils and sierozems",
+    "Loukkos": "Mostly gleysols and brunified",
+    "Rharb plain": "Gleysols and vertisols",
+    "Central plateau": "In forested areas, soils are brown associated with lithosols and regosols. Elsewhere (Zaer), vertisols and gleysols dominate",
+    "Plains and plateaux of north of the Atlas": "Lithosols (Rehamnas, Jebilete), sierozems associated with lithosols",
+    "Argan zone": "Soils are mostly lithosols and regosols, associated with fluvisols and saline soils on lowlands",
+    "Presaharan soils": "Lithosols and regosols in association with sierozems and regs",
+    "Saharan zone": "Yermosols, associated with sierozems, lithosols, and saline soils"
+}
+
 def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.json()
@@ -32,7 +49,7 @@ def translate(text,source="English",target="Moroccan Arabic"):
 
 
 # Function to generate a response from the chatbot
-def generate_response(user_input):
+def generate_response(user_input,region):
 
     user_input_translated = str(translate(user_input, "Moroccan Arabic", "English"))
     name = 'Mohammed'
@@ -51,7 +68,7 @@ def generate_response(user_input):
     <s> [INST] You are an agriculture expert, and my name is {name} Given the following informations, prevailing weather conditions, specific land type, chosen type of agriculture, and soil composition of a designated area, answer the question below
 Location: {location},
 Current Month : {date}
-land type: {soil_type}
+land type: {soil_types[region]}
 humidity: {humidity}
 weather: {weather}
 temperature: {temp}
@@ -116,6 +133,7 @@ def main():
     # sidebar_bg('bg.jpg')
     # Layout of input/response containers
     input_container = st.container()
+    selected_region = st.selectbox("Choose a region:", list(soil_types.keys()))
 
     if st.button("Clear Chat"):
         st.session_state['past'] = []
@@ -137,7 +155,7 @@ def main():
     ## Conditional display of AI generated responses as a function of user provided prompts
     with response_container:
         if user_input:
-            response = generate_response(user_input)
+            response = generate_response(user_input,str(selected_region))
             st.session_state.past.append(user_input)
             st.session_state.generated.append(response)