Update app.py

app.py

@@ -1546,7 +1546,7 @@ def save_image_old2(image, filename):
 
 # Now filename length protected for linux and windows filename lengths
 def save_image(image, filename):
-    max_filename_length = min(os.pathconf('/', 'PC_NAME_MAX'), 255, 260)  # Linux: 255, Windows: 260 (including path)
+    max_filename_length = 250
     filename_stem, extension = os.path.splitext(filename)
     truncated_stem = filename_stem[:max_filename_length - len(extension)] if len(filename) > max_filename_length else filename_stem
     filename = f"{truncated_stem}{extension}"
@@ -1557,7 +1557,15 @@ def save_image(image, filename):
 def extract_boldface_terms(text):
     return re.findall(r'\*\*(.*?)\*\*', text)
 
-def process_image(image_input):
+def extract_title(text):
+    boldface_terms = re.findall(r'\*\*(.*?)\*\*', text)
+    if boldface_terms:
+        title = ' '.join(boldface_terms)
+    else:
+        title = re.sub(r'[^a-zA-Z0-9_\-]', ' ', text[-200:])
+    return title[-200:]
+
+def process_image(image_input, user_prompt):
     if image_input:
         st.markdown('Processing image:  ' + image_input.name )
         if image_input:
@@ -1567,7 +1575,7 @@ def process_image(image_input):
                 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": "text", "text": user_prompt},
                         {"type": "image_url", "image_url": {
                             "url": f"data:image/png;base64,{base64_image}"}
                         }
@@ -1589,9 +1597,10 @@ def process_image(image_input):
                 f.write(image_response)
 
             # Extract boldface terms from image_response then autoname save file
-            boldface_terms = extract_boldface_terms(image_response)
+            #boldface_terms = extract_boldface_terms(image_response)
+            boldface_terms = extract_title(image_response)
             filename_stem, extension = os.path.splitext(image_input.name)
-            filename_img = f"{filename_stem}_{' '.join(boldface_terms)}{extension}"
+            filename_img = f"{filename_stem}  {''.join(boldface_terms)}{extension}"
             newfilename = save_image(image_input, filename_img)
             filename_md = newfilename.replace('.png', '.md')
             create_file(filename_md, '', image_response, True)
@@ -1604,34 +1613,6 @@ def save_imageold(image_input, filename_txt):
         f.write(image_input.getbuffer())
     return image_input.name
 
-def process_imageold(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)
-        
-        filename_txt = generate_filename(image_response, "md")  # Save markdown on image AI output from gpt4o
-        create_file(filename_txt, image_response, '', True)          #create_file() # create_file()  3 required positional arguments: 'filename', 'prompt', and 'response'
-
-        filename_txt = generate_filename(image_response, "png")
-        save_image(image_input, filename_txt)     # Save copy of image with new filename   
-        #st.rerun() # rerun to show new image and new markdown files
-
-        return image_response
-
 
 def process_audio(audio_input):
     if audio_input:
@@ -1747,8 +1728,10 @@ def main():
         if (text_input > ''):
             textResponse = process_text(text_input)
     elif option == "Image":
+        text = "Help me understand what is in this picture and list ten facts as markdown outline with appropriate emojis that describes what you see."
+        text_input = st.text_input(label="Enter text prompt to use with Image context.", value=text)
         image_input = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-        image_response = process_image(image_input)
+        image_response = process_image(image_input, text_input)