Update app.py

app.py

@@ -8,32 +8,38 @@ TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
 # Streamlit UI
 st.title("Movie Review Sentiment Analysis")
 
-# 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:", "")
+# 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 prompts:", "")
 
 # 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 = few_shot_prompt
+else:
+    prompt = ""
+
 # Make a request to Together API if a prompt is provided
-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}"
-        })
+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'])
-    else:
-        st.warning("Please enter a prompt.")
+    # Display the output
+    st.text("Sentiment Analysis Result:")
+    st.text(response.json()['output']['choices'][0]['text'])