Update app.py

app.py

@@ -397,7 +397,7 @@ def infer(crop_image_path,full_image_path,state,language,task_type=None):
     
     elif task_type=="task 3":
         item_gallery_output.append("recomendation_pic/3.8.png")
-        item_gallery_output.append("recomendation_pic/basket-2.jpg")
+        item_gallery_output.append("recomendation_pic/basket-2.png")
         input_image = Image.open(full_image_path).convert("RGB")
         input_features = extract_features_siglip(input_image.convert("RGB"))
         input_features = input_features.detach().cpu().numpy()
@@ -999,6 +999,27 @@ async def upload_callback(image_input,state, log_state, task_type, visual_chatgp
             year="12th century (Song Dynasty)"
             material="Chinese painting"
             gender="male"
+        
+        elif task_type=="task 1":
+            name ="The Ambassadors"
+            artist ="Hans Holbein the Younger"
+            year = "1533 (Northern Renaissance)"
+            material= "Realism"
+            gender = "male"
+        
+        elif task_type=="task 2":
+            name ="The Ambassadors"
+            artist ="Hans Holbein the Younger"
+            year = "1533 (Northern Renaissance)"
+            material= "Realism"
+            gender = "male"
+        
+        elif task_type=="task 2":
+            name = "The Football Players"
+            artist= "Albert Gleizes"
+            year= "1912 (Cubism)"
+            material= "Cubism",
+            gender= "male"
         else:
             parsed_data = get_gpt_response(openai_api_key, new_image_path,"Please provide the name, artist, year of creation (including the art historical period), and painting style used for this painting. Return the information in dictionary format without any newline characters. Format as follows: { \"name\": \"Name of the painting\", \"artist\": \"Name of the artist\", \"year\": \"Year of creation (Art historical period)\", \"style\": \"Painting style used in the painting\",\"gender\": \"The gender of the author\"}")
             print(parsed_data)
@@ -1675,7 +1696,7 @@ async def texttospeech(text, language,gender='female'):
         return None
 
 # give the reason of recommendation
-async def item_associate(new_crop,openai_api_key,language,autoplay,length,log_state,sort_score,narritive,state,history_log, evt: gr.SelectData):
+async def item_associate(new_crop,openai_api_key,language,autoplay,length,log_state,sort_score,narritive,state, evt: gr.SelectData):
     persona=naritive_mapping[narritive]
     rec_path=evt._data['value']['image']['path']
     index="Item Recommendation Picture "+ str(evt.index)
@@ -1684,7 +1705,7 @@ async def item_associate(new_crop,openai_api_key,language,autoplay,length,log_st
     
     image_paths=[new_crop,rec_path]
 
-    result=get_gpt_response(openai_api_key, image_paths, prompt,history_log)
+    result=get_gpt_response(openai_api_key, image_paths, prompt)
     print("recommend result",result)
     state += [(None, f"{result}")]
     log_state += [("User wants to know object recomendation reason", None)]
@@ -1700,14 +1721,14 @@ async def item_associate(new_crop,openai_api_key,language,autoplay,length,log_st
     return state,state,audio_output,log_state,index,gr.update(value=[])
 
 
-async def style_associate(image_path,openai_api_key,language,autoplay,length,log_state,sort_score,narritive,state,history_log,evt: gr.SelectData):
+async def style_associate(image_path,openai_api_key,language,autoplay,length,log_state,sort_score,narritive,state,evt: gr.SelectData):
     persona=naritive_mapping[narritive]
     rec_path=evt._data['value']['image']['path']
     index="Style Recommendation Picture "+str(evt.index)
     print("rec_path",rec_path)
     prompt=recommendation_prompt[1][persona].format(language=language,length=length)
     image_paths=[image_path,rec_path]
-    result=get_gpt_response(openai_api_key, image_paths, prompt,history_log)
+    result=get_gpt_response(openai_api_key, image_paths, prompt )
     print("recommend result",result)
     state += [(None, f"{result}")]
     log_state += [("User wants to know style recomendation reason", None)]
@@ -2284,7 +2305,7 @@ def create_ui():
 
         gallery_result.select(
             item_associate,
-            inputs=[new_crop_save_path,openai_api_key,language,auto_play,length,log_state,sort_rec,naritive,recomended_state,history_log],
+            inputs=[new_crop_save_path,openai_api_key,language,auto_play,length,log_state,sort_rec,naritive,recomended_state],
             outputs=[recommend_bot,recomended_state,output_audio,log_state,pic_index,recommend_score],
             
             
@@ -2292,7 +2313,7 @@ def create_ui():
         
         style_gallery_result.select(
             style_associate,
-            inputs=[image_path,openai_api_key,language,auto_play,length,log_state,sort_rec,naritive,recomended_state,history_log],
+            inputs=[image_path,openai_api_key,language,auto_play,length,log_state,sort_rec,naritive,recomended_state],
             outputs=[recommend_bot,recomended_state,output_audio,log_state,pic_index,recommend_score],