Update app.py

app.py

@@ -1602,7 +1602,7 @@ def process_text():
         )
         st.write("Assistant: " + completion.choices[0].message.content)
 
-def process_image(image_input):
+def process_image_old_05152024(image_input):
     if image_input:
         base64_image = base64.b64encode(image_input.read()).decode("utf-8")
         response = client.chat.completions.create(
@@ -1620,6 +1620,51 @@ def process_image(image_input):
         )
         st.markdown(response.choices[0].message.content)
 
+
+def save_image(image_input, image_response):
+    if image_input and image_response:
+        # Extract emojis and first two words from each markdown line
+        lines = image_response.split("\n")
+        filename_parts = []
+        for line in lines:
+            if line.startswith("- "):
+                emoji = re.search(r'(\p{Extended_Pictographic})', line)
+                words = re.findall(r'\b\w+\b', line)
+                if emoji and len(words) >= 2:
+                    filename_parts.append(f"{emoji.group(1)}_{words[1]}_{words[2]}")
+
+        # Create the filename by concatenating the extracted parts
+        filename = "_".join(filename_parts)[:50]  # Limit filename length to 50 characters
+        filename = f"{filename}.png"
+
+        # Save the image with the new filename
+        with open(filename, "wb") as f:
+            f.write(image_input.getbuffer())
+
+        return filename
+
+def process_image(image_input):
+    if image_input:
+        base64_image = base64.b64encode(image_input.read()).decode("utf-8")
+        response = client.chat.completions.create(
+            model=MODEL,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant that responds in Markdown."},
+                {"role": "user", "content": [
+                    {"type": "text", "text": "Help me understand what is in this picture and list ten facts as markdown outline with appropriate emojis that describes what you see."},
+                    {"type": "image_url", "image_url": {
+                        "url": f"data:image/png;base64,{base64_image}"}
+                    }
+                ]}
+            ],
+            temperature=0.0,
+        )
+        image_response = response.choices[0].message.content
+        st.markdown(image_response)
+
+        # Save the image with a new filename based on the response
+        save_image(image_input, image_response)        
+
 def process_audio(audio_input):
     if audio_input:
         transcription = client.audio.transcriptions.create(
@@ -1741,14 +1786,10 @@ if __name__ == "__main__":
     main()
         
 display_glossary_grid(roleplaying_glossary)  # Word Glossary Jump Grid - Dynamically calculates columns based on details length to keep topic together
-            
-
 num_columns_video=st.slider(key="num_columns_video", label="Choose Number of Video Columns", min_value=1, max_value=15, value=4)
 display_videos_and_links(num_columns_video)   # Video Jump Grid
-
 num_columns_images=st.slider(key="num_columns_images", label="Choose Number of Image Columns", min_value=1, max_value=15, value=4)
 display_images_and_wikipedia_summaries(num_columns_images)   # Image Jump Grid
-
 num_columns_text=st.slider(key="num_columns_text", label="Choose Number of Text Columns", min_value=1, max_value=15, value=4)
 display_buttons_with_scores(num_columns_text)  # Feedback Jump Grid