Update app.py

app.py

@@ -88,17 +88,21 @@ def model_function(input_data):
 
 # Streamlit UI code
 def key_value_input(container, label, key, value, index):
-    common_headers = [
-        "Host", "User-Agent", "Accept", "Accept-Language", 
-        "Accept-Encoding", "Connection", "Referer", "Cookie", 
-        "Authorization", "Cache-Control", "Content-Type"
-    ]
     col1, col2 = container.columns(2)
 
-    # Adding a selectbox for common headers
-    key_input = col1.selectbox(f"{label} Key {index}", options=[''] + common_headers, index=common_headers.index(key) if key in common_headers else 0, key=f"{label}_key_{index}")
-    value_input = col2.text_input(f"{label} Value {index}", value, key=f"{label}_value_{index}")
+    # Use a simple text_input for keys in Request Body and Response Object
+    if label in ["Request Body", "Response Object"]:
+        key_input = col1.text_input(f"{label} Key {index}", key, key=f"{label}_key_{index}")
+    else:
+        # Existing logic for Request Header with common headers
+        common_headers = [
+            "Host", "User-Agent", "Accept", "Accept-Language", 
+            "Accept-Encoding", "Connection", "Referer", "Cookie", 
+            "Authorization", "Cache-Control", "Content-Type","API-Key"
+        ]
+        key_input = col1.selectbox(f"{label} Key {index}", options=[''] + common_headers, index=common_headers.index(key) if key in common_headers else 0, key=f"{label}_key_{index}")
 
+    value_input = col2.text_input(f"{label} Value {index}", value, key=f"{label}_value_{index}")
     return key_input, value_input
 
 def dynamic_key_value_pairs(label):