Update app.py

app.py

@@ -6,44 +6,41 @@ import os
 TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
 
 # Streamlit UI
-st.title("Zero Shot, Single-Shot, Few Shots Prompting Showcase")
+st.title("Movie Review Sentiment Analysis")
 
-# User input for prompt and examples
-prompt_and_examples = st.text_area("Enter your prompt and examples:", """\
-  Label the sentences as either "positive", "negative", "mixed", or "neutral":
-
-  Sentence: I can say that there isn't anything I would change.
-  Label: positive
-
-  Sentence: I'm not sure about this.
-  Sentence: neutral
-
-  Sentence: I liked some parts but I didn't like other parts.
-  Label: mixed
-
-  Sentence: I think the background image could have been better.
-  Label: negative
-
-  Sentence: I really like it.
-  Label:""")
+# 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):", "")
 
 # Together API endpoint
-endpoint = 'https://api.together.xyz/inference'
-
-# Make a request to Together API
-res = requests.post(endpoint, json={
-    "model": 'togethercomputer/RedPajama-INCITE-7B-Base',
-    "prompt": prompt_and_examples,
-    "top_p": 1,
-    "top_k": 40,
-    "temperature": 0.8,
-    "max_tokens": 1,
-    "repetition_penalty": 1,
-}, headers={
-    "Authorization": f"Bearer {TOGETHER_API_KEY}",
-    "User-Agent": "<YOUR_APP_NAME>"
-})
-
-# Display the output
-st.text("Output:")
-st.text(res.json()['output']['choices'][0]['text'])
+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}"
+    })
+
+    # Display the output
+    st.text("Sentiment Analysis Result:")
+    st.text(response.json()['output']['choices'][0]['text'])