Update app.py

app.py

@@ -10,6 +10,7 @@ code = os.environ.get("code")
 model_llm = os.environ.get("model")
 content = os.environ.get("content")
 state = os.environ.get("state")
+system = os.environ.get("system")
 data = None
 model = None
 image = None
@@ -49,7 +50,7 @@ def classify(Textbox, Image, Textbox2, Textbox3):
         Textbox2_edited = list(Textbox2_edited)
         Textbox2_edited.append(Textbox)
         messages = [
-            {"role": "system", "content": content},
+            {"role": "system", "content": system},
         ]
         print("Messages",messages)
         
@@ -75,6 +76,7 @@ def classify(Textbox, Image, Textbox2, Textbox3):
             time.sleep(1)
             print("\nWays to dispose of this waste: " + max_label)
             messages.append(
+                {"role": "user", "content": content + " " +max_label},
                 {"role": "user", "content": Textbox},
             )