Update app.py

app.py

@@ -25,7 +25,7 @@ llm = Llama(
 ) 
 
 
-def run_pic(image):
+# def run_pic(image):
     image = Image.open(image[0])
     task_prompt = "<DESCRIPTION>"
     prompt = task_prompt + "Describe this image in great detail."
@@ -56,16 +56,18 @@ def generate_text(
     in_text = message['text']
     in_files = message['files']
     output=""
+    picoutput=""
+    history_prompt=""
     if in_files:
-        output=run_pic(in_files)
-        yield output
+        picoutput=f"sends a picture that contains the following:  {run_pic(in_files)}"
+        yield picoutput
     else:
         temp = ""
-        input_prompt = f'{system_message}'
-        # for interaction in history:
-            # input_prompt = input_prompt + str(interaction[0]) +  str(interaction[1])
-
-        input_prompt = input_prompt + "  " + str(in_text) 
+        system_prompt = f'{system_message}'
+        for interaction in history:
+            history_prompt += str(interaction[0]) +  str(interaction[1])
+        input_prompt = system_prompt + "  " + history_prompt + "  " + str(in_text) 
+        print(input_prompt)
         output = llm(
             input_prompt,
             temperature=temperature,
@@ -116,4 +118,4 @@ demo = gr.ChatInterface(
 
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0")
+    demo.launch()