Update app.py

app.py

@@ -8,39 +8,32 @@ TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
 # Streamlit UI
 st.title("Movie Review Sentiment Analysis")
 
-# User input for movie review
-zero_shot_prompt = st.text_area("Enter zero-shot prompt:", "")
-one_shot_prompt = st.text_area("Enter one-shot prompt:", "")
-few_shot_prompt = st.text_area("Enter few-shot prompt (comma-separated options):", "")
+# User input for prompting technique
+prompting_technique = st.selectbox("Select prompting technique:", ["Zero-shot", "One-shot", "Few-shot"])
+
+# Dynamic textbox based on user selection
+prompt = st.text_area(f"Enter {prompting_technique} prompt:", "")
 
 # Together API endpoint
 endpoint = 'https://api.together.xyz/inference' 
 
-# Determine the prompt based on user input
-if st.button("Analyze Zero-shot"):
-    prompt = zero_shot_prompt
-elif st.button("Analyze One-shot"):
-    prompt = one_shot_prompt
-elif st.button("Analyze Few-shot"):
-    prompt_options = few_shot_prompt.split(',')
-    prompt = '\n'.join([f"Option {i + 1}: {option.strip()}" for i, option in enumerate(prompt_options)])
-else:
-    prompt = ""
-
 # Make a request to Together API if a prompt is provided
-if prompt:
-    response = requests.post(endpoint, json={
-        "model": 'togethercomputer/RedPajama-INCITE-7B-Base',
-        "prompt": prompt,
-        "top_p": 1,
-        "top_k": 40, 
-        "temperature": 0.8,
-        "max_tokens": 1,
-        "repetition_penalty": 1,
-    }, headers={
-        "Authorization": f"Bearer {TOGETHER_API_KEY}"
-    })
+if st.button(f"Analyze {prompting_technique}"):
+    if prompt:
+        response = requests.post(endpoint, json={
+            "model": 'togethercomputer/RedPajama-INCITE-7B-Base',
+            "prompt": prompt,
+            "top_p": 1,
+            "top_k": 40, 
+            "temperature": 0.8,
+            "max_tokens": 1,
+            "repetition_penalty": 1,
+        }, headers={
+            "Authorization": f"Bearer {TOGETHER_API_KEY}"
+        })
 
-    # Display the output
-    st.text("Sentiment Analysis Result:")
-    st.text(response.json()['output']['choices'][0]['text'])
+        # Display the output
+        st.text("Sentiment Analysis Result:")
+        st.text(response.json()['output']['choices'][0]['text'])
+    else:
+        st.warning("Please enter a prompt.")