spliting files

app.py

@@ -1,140 +1,93 @@
 
-"""ChatGPT 4 was used to set up a structure for this function"""
+"""
+ChatGPT 4 was used to set up a structure for this project, but then everything was pretty much rewritten as project evolved.
+We strongly encourage you to integrate GenAI tools into your work. Doing so can not only jump-start your process and provide solutions when you hit a roadblock, but it also offers a fantastic opportunity to deepen your understanding of AI's potential and expand your creative and technical horizons
+"""
 
 
 import gradio as gr
-import os
-import langchain_openai
-import langchain_core
-import json
-from langchain_openai import ChatOpenAI
-from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.output_parsers import StrOutputParser
+import utils
 
-def load_input_fields(filepath):
-    """
-    Load the input fields from a JSON file.
+# Load input fields data from the JSON file
+input_fields = utils.load_input_fields("input_fields.json")
 
-    Args:
-    - filepath (str): The path to the JSON file containing the input fields.
-
-    Returns:
-    - dict: A dictionary containing the input fields.
-    """
-
-    with open(filepath, "r") as file:
-        input_fields = json.load(file)
-
-    return input_fields
-
-def create_html_string(input_text, highlight_color = "green", container_style = "border: 2px solid black; padding: 2px; font-size: 16px;" ):
-    """
-    Create a HTML string with specific styles applied to highlighted text within square brackets.
-
-    Args:
-    - input_text (str): The input text with portions to be highlighted within square brackets.
-    - optional: highlight_color (str): Color for the highlighted text (e.g., "green").
-    - optional: container_style (str): Any css for inline styling (e.g,, "border: 2px solid black;")
-
-    Returns:
-    - str: A HTML string with the applied styles.
-    """
-
-    # Replace the highlighted text with HTML span elements for styling
-    highlighted_text = input_text.replace("[", f'<span style="color:{highlight_color}; font-weight: bold;">[').replace("]", "]</span>")
-
-    # Construct the full HTML string with the provided styles
-    html_string = f'<p style="{container_style}">{highlighted_text}</p>'
-
-    return html_string
-
-def extract_names(objects):
-    return [obj['name'] for obj in objects if 'name' in obj]
-
-def format_to_markdown(objects):
-    # Skip None objects
-    formatted_list = [
-        f"> * **{obj.get('name', 'No Name')}** - {obj.get('description', 'No Description')}"
-        for obj in objects if obj is not None and obj["name"] != "None"
-    ]
-    return '\n'.join(formatted_list)
-
-def enhance_subject(subject, details):
-  prompt = ChatPromptTemplate.from_messages([
-      ("system", "Generate a clear and concise subject, then provide additional details using descriptive language. Ensure the response is specific and avoids ambiguity or contradictions. The subject should inspire an engaging photo that tells a story. Remove any unnecessary information and don't add any punctuation at the end of the subject."),
-      ("user", "The main subject is {subject} {details}.")
-    ])
-  output_parser = StrOutputParser()
-  model = ChatOpenAI(model="gpt-3.5-turbo")
-  chain = ( prompt
-    | model
-    | output_parser
-  )
-  result = chain.invoke({"subject": subject, "details": details})
-  return result
-
-
-input_fields = load_input_fields("input_fields.json")
-# Models
 models = input_fields.get("models")
 # Styles
 photography_styles = input_fields.get("photography_styles")
+
 # Environment
 weather_conditions = input_fields.get("weather_conditions")
 time_conditions = input_fields.get("time_conditions")
 moods_atmospheres = input_fields.get("moods_atmospheres")
 lighting_types = input_fields.get("lighting_types")
+
 # Camera and Composition
 cameras = input_fields.get("cameras")
 camera_angles = input_fields.get("camera_angles")
 shot_types =  input_fields.get("shot_types")
 lens_filters = input_fields.get("lens_filters")
+
 # Midjourney Only Parameters
 aspect_ratio_variations = input_fields.get("aspect_ratio_variations")
 temperature_variations = input_fields.get("temperature_variations")
 
 welcome_text="[Photo Style] of [Subject], set in a [Environment — time, weather, etc.] with a [Mood]. Include [Additional Details]. Emulate a [Specific Camera] with a [Lens Filter] and use [Shot Type]. Compose at [Camera Angle] from [Camera Position] perspective. Use [Lighting Conditions]."
 container_style = "border: 2px solid black; padding: 2rem; font-size: 16px;"
-html_formated_text=create_html_string(welcome_text, container_style=container_style)
+
+html_formated_text=utils.create_html_string(welcome_text, container_style=container_style)
 
 def display_camera_info(camera_type):
-  find_filter_by_name = lambda name: next((filter for filter in cameras if filter['name'] == name), None)
-  markdown_text = format_to_markdown([find_filter_by_name(camera_type)])
-  return gr.Markdown(markdown_text, visible=True)
+   return utils.display_info(cameras, camera_type)
 
 def display_light_info(lighting_type):
-  find_filter_by_name = lambda name: next((filter for filter in lighting_types if filter['name'] == name), None)
-  markdown_text = format_to_markdown([find_filter_by_name(lighting_type)])
-  return gr.Markdown(markdown_text, visible=True)
-  
-def display_lens_info(lens_filter):
-  find_filter_by_name = lambda name: next((filter for filter in lens_filters if filter['name'] == name), None)
-  markdown_text = format_to_markdown([find_filter_by_name(lens_filter)])
-  return gr.Markdown(markdown_text, visible=True)
+  return utils.display_info(lighting_types, lighting_type)
 
+def display_lens_info(lens_filter):
+  return utils.display_info(lens_filters, lens_filter)
+  
 def display_ar_info(aspect_ratio):
-  find_filter_by_name = lambda name: next((filter for filter in aspect_ratio_variations if filter['name'] == name), None)
-  markdown_text = format_to_markdown([find_filter_by_name(aspect_ratio)])
-  return gr.Markdown(markdown_text, visible=True)
+  return utils.display_info(aspect_ratio_variations, aspect_ratio)
 
 def display_temperature_info(temperature):
-  find_filter_by_name = lambda name: next((filter for filter in temperature_variations if filter['name'] == name), None)
-  markdown_text = format_to_markdown([find_filter_by_name(temperature)])
-  return gr.Markdown(markdown_text, visible=True)
+  return utils.display_info(temperature_variations, temperature)
 
 def display_shot_info(shot_type):
-  find_filter_by_name = lambda name: next((filter for filter in shot_types if filter['name'] == name), None)
-  markdown_text = format_to_markdown([find_filter_by_name(shot_type)])
-  return gr.Markdown(markdown_text, visible=True)
-def convert_to_ar_format(input_str):
-    start = input_str.find('(') + 1
-    end = input_str.find(')', start)
-    ratio_part = input_str[start:end]
-    x, y = ratio_part.split(':')
-    return f"--ar {x}:{y}"
-def convert_to_temp_format(temperature):
-  find_filter_by_name = lambda name: next((filter for filter in temperature_variations if filter['name'] == name), None)
-  return find_filter_by_name(temperature)['param']
+  return utils.display_info(shot_types, shot_type)
+
+def compile_prompt(photo_style, subject, enhanced_prompt, weather, time, mood, details, camera_type, camera_angle, shot_type, lens_filter, lighting, model, aspect_ratio, temperature):
+    resulting_prompt = ""
+    if photo_style and photo_style != "None":
+        resulting_prompt += f"{photo_style} of "
+    if subject and (enhanced_prompt is None or enhanced_prompt == ""):
+        resulting_prompt += f"{subject}, "
+    if enhanced_prompt and enhanced_prompt != "":
+        resulting_prompt += f"{enhanced_prompt}, "
+    if weather != "None" or time != "None":
+        resulting_prompt += "set  "
+    if weather and weather != "None":
+        resulting_prompt += f" on a {weather} day "
+    if time and time!= "None":
+        resulting_prompt += f"at {time} "
+    if mood and mood != "None":
+        resulting_prompt += f"with a {mood} mood. "
+    if details and (enhanced_prompt is None or enhanced_prompt == ""):
+        resulting_prompt += f"Include {details}. "
+    if camera_type and camera_type != "None":
+        resulting_prompt += f"Shot on {camera_type} "
+    if lens_filter and lens_filter != "None":
+        resulting_prompt += f"with a {lens_filter} lens filter "
+    if shot_type and shot_type != "None":
+        resulting_prompt += f"and use a {shot_type} shot type. "
+    if camera_angle and camera_angle != "None":
+        resulting_prompt += f"Compose at {camera_angle} "
+    if lighting and lighting != "None":
+        resulting_prompt += f"Use {lighting} lighting conditions "
+    if model == "Midjourney" and aspect_ratio != "None":
+        resulting_prompt += f"{utils.find_filter_by_name(aspect_ratio_variations, aspect_ratio)['param']} "
+    if model == "Midjourney" and temperature != "None":
+        resulting_prompt += f"{utils.find_filter_by_name(temperature_variations, temperature)['param']} "
+    return resulting_prompt
+
 
 with gr.Blocks() as demo:
   isFrog = gr.State(False)
@@ -157,23 +110,6 @@ with gr.Blocks() as demo:
       """
   )
 
-  def enhance_pipeline(isFrog, subject, details):
-    if isFrog and (subject or details):
-      result = enhance_subject(subject, details)
-      return [gr.Textbox(visible=False), gr.Button(visible=False), gr.TextArea(visible=True, value=result)]
-    elif (subject or details) and not isFrog:
-      return [gr.Textbox(visible=True), gr.Button(visible=True), gr.TextArea(visible=False)]
-    else:
-      return [gr.Textbox(visible=False), gr.Button(visible=False), gr.TextArea(visible=False)]
-
-  
-  def authenticate(pwd_input, subject, details):
-    if pwd_input == os.environ.get("MAGIC_WORD"):
-      result = enhance_subject(subject, details)
-      return [gr.TextArea(visible=True, value=result), True, gr.Textbox(visible=False), gr.Button(visible=False)]
-    else:
-      raise gr.Error("You are not from our pond! Use your own LLM to add some juice to your prompt.")
-
   with gr.Row():
     with gr.Column():
       ui_subject = gr.Textbox(label="Subject of prompt", placeholder="Type your subject", info="Type the subject of the prompt you are creating")
@@ -181,26 +117,23 @@ with gr.Blocks() as demo:
       ui_details = gr.Textbox(label="Details", placeholder="more details", info="Add more details to the prompt")
   with gr.Row():
     with gr.Column():
-      enhance_button = gr.Button(value="🪄 Do Magic", variant="primary", size="lg")
+      enhance_button = gr.Button(value="🪄 Enhance Subject", variant="primary", size="lg")
     with gr.Column():
       pwd_input = gr.Textbox(label="Do you know the magic word?", visible=False)
       confirm_btn = gr.Button(value="Confirm Pwd", visible=False)
       enhanced_prompt = gr.TextArea(label="Enhanced Prompt", show_copy_button=True, interactive=True, visible=False, lines=3)
   
   enhance_button.click(
-    enhance_pipeline,
+    utils.enhance_pipeline,
     inputs=[isFrog, ui_subject, ui_details], 
     outputs=[pwd_input, confirm_btn, enhanced_prompt]
   )
   
-  def clearInput():
-    return ""
-
   confirm_btn.click(
-    authenticate, 
+    utils.authenticate, 
     inputs=[pwd_input, ui_subject, ui_details],
     outputs=[enhanced_prompt,isFrog, pwd_input, confirm_btn]
-  ).then(clearInput, outputs=pwd_input)
+  ).then(utils.clearInput, outputs=pwd_input)
   
   gr.Markdown(
       """
@@ -213,8 +146,6 @@ with gr.Blocks() as demo:
       with gr.Column():
         ui_photography_style = gr.Dropdown(choices=photography_styles, value="None", interactive=True, allow_custom_value=True, label="Photography Style", info="Select a type of photography or add your own.")
 
-
-  
   gr.Markdown(
       """
        ## Parameters
@@ -235,7 +166,7 @@ with gr.Blocks() as demo:
         ui_mood = gr.Dropdown(choices=moods_atmospheres, value="None", interactive=True, allow_custom_value=True, label="Mood/Atmosphere", info="Change the tone of your photos or expressions. Note: it will be more effective also adding these descriptors in your subject.")
     with gr.Row():
       with gr.Column():
-        ui_lighting = gr.Dropdown(choices=extract_names(lighting_types), value="None", interactive=True, allow_custom_value=True, label="Lighting", info="Indicate how you want your subject or scene’s lighting to be")
+        ui_lighting = gr.Dropdown(choices=utils.extract_names(lighting_types), value="None", interactive=True, allow_custom_value=True, label="Lighting", info="Indicate how you want your subject or scene’s lighting to be")
       with gr.Column():
         lighting_info = gr.Markdown(visible=False)
       ui_lighting.change(
@@ -251,7 +182,7 @@ with gr.Blocks() as demo:
           """
           Film Camera: 🎞️ | DSLR Camera: 📸 | Special Effect Camera: ✨📷
           """)
-        ui_camera_type = gr.Radio(choices=extract_names(cameras), value="None", interactive=True, label="Camera Specs", info="Emulate what you might see if you took the photo from one of these cameras.")
+        ui_camera_type = gr.Radio(choices=utils.extract_names(cameras), value="None", interactive=True, label="Camera Specs", info="Emulate what you might see if you took the photo from one of these cameras.")
       with gr.Column():
         camera_info_block = gr.Markdown(visible=False)
       ui_camera_type.change(
@@ -270,7 +201,7 @@ with gr.Blocks() as demo:
       with gr.Column():
         ui_camera_angle = gr.Radio(choices=camera_angles, value="None", interactive=True, label="Camera Angles", info="Select an angle")
       with gr.Column():
-        ui_shot_type = gr.Radio(choices=extract_names(shot_types), value="None", interactive=True, label="Shot Type", info="Select zoom type")
+        ui_shot_type = gr.Radio(choices=utils.extract_names(shot_types), value="None", interactive=True, label="Shot Type", info="Select zoom type")
       with gr.Column():
         shot_type_info = gr.Markdown(visible=False)
       ui_shot_type.change(
@@ -280,7 +211,7 @@ with gr.Blocks() as demo:
         )
     with gr.Row("Lens Filter"):
       with gr.Column():
-        ui_lens_filter = gr.Radio(choices=extract_names(lens_filters), value="None", interactive=True, label="Lens Filter", info="Give some slight nuance to the tone of your photographs")
+        ui_lens_filter = gr.Radio(choices=utils.extract_names(lens_filters), value="None", interactive=True, label="Lens Filter", info="Give some slight nuance to the tone of your photographs")
       with gr.Column():
         lens_filter_info = gr.Markdown(visible=False)
       ui_lens_filter.change(
@@ -291,7 +222,7 @@ with gr.Blocks() as demo:
   with gr.Tab(label="Midjourney Only"):
     with gr.Row("Aspect Ratio"):
       with gr.Column():
-        ui_aspect_ratio = gr.Radio(choices=extract_names(aspect_ratio_variations),value="None", interactive=True, label="Aspect Ratio", info="Select the aspect ratio of your photograph")
+        ui_aspect_ratio = gr.Radio(choices=utils.extract_names(aspect_ratio_variations),value="None", interactive=True, label="Aspect Ratio", info="Select the aspect ratio of your photograph")
       with gr.Column():
         aspect_ratio_info = gr.Markdown(visible=False)
       ui_aspect_ratio.change(
@@ -301,7 +232,7 @@ with gr.Blocks() as demo:
         )
     with gr.Row("Aspect Ratio"):
       with gr.Column():
-        ui_temperature = gr.Radio(choices=extract_names(temperature_variations), value="None", interactive=True, label="Temperature", info="Indicate how much you want your generated photo to stay true to your prompt. Only select these if you want variation, are concepting, or are open to results that can pleasantly surprise you!")
+        ui_temperature = gr.Radio(choices=utils.extract_names(temperature_variations), value="None", interactive=True, label="Temperature", info="Indicate how much you want your generated photo to stay true to your prompt. Only select these if you want variation, are concepting, or are open to results that can pleasantly surprise you!")
       with gr.Column():
         temperature_info = gr.Markdown(visible=False)
       ui_temperature.change(
@@ -316,40 +247,6 @@ with gr.Blocks() as demo:
     with gr.Column(scale=3):
       result_prompt=gr.TextArea(label="Compiled Prompt", show_copy_button=True, interactive=True)
 
-  def compile_prompt(photo_style, subject, enhanced_prompt, weather, time, mood, details, camera_type, camera_angle, shot_type, lens_filter, lighting, model, aspect_ratio, temperature):
-    resulting_prompt = ""
-    if photo_style and photo_style != "None":
-      resulting_prompt += f"{photo_style} of "
-    if subject and (enhanced_prompt is None or enhanced_prompt == ""):
-      resulting_prompt += f"{subject}, "
-    if enhanced_prompt and enhanced_prompt != "":
-      resulting_prompt += f"{enhanced_prompt}, "
-    if weather != "None" or time != "None":
-      print(weather, time)
-      resulting_prompt += "set  "
-    if weather and weather != "None":
-      resulting_prompt += f" on a {weather} day "
-    if time and time!= "None":
-      resulting_prompt += f"at {time} "
-    if mood and mood != "None":
-      resulting_prompt += f"with a {mood} mood. "
-    if details and (enhanced_prompt is None or enhanced_prompt == ""):
-      resulting_prompt += f"Include {details}. "
-    if camera_type and camera_type != "None":
-      resulting_prompt += f"Shot on {camera_type} "
-    if lens_filter and lens_filter != "None":
-      resulting_prompt += f"with a {lens_filter} lens filter "
-    if shot_type and shot_type != "None":
-      resulting_prompt += f"and use a {shot_type} shot type. "
-    if camera_angle and camera_angle != "None":
-      resulting_prompt += f"Compose at {camera_angle} "
-    if lighting and lighting != "None":
-      resulting_prompt += f"Use {lighting} lighting conditions "
-    if model == "Midjourney" and aspect_ratio != "None":
-      resulting_prompt += f"{convert_to_ar_format(aspect_ratio)} "
-    if model == "Midjourney" and temperature != "None":
-      resulting_prompt += f"{convert_to_temp_format(temperature)} "
-    return resulting_prompt
   
   compile_button.click(
     compile_prompt,
@@ -373,9 +270,6 @@ with gr.Blocks() as demo:
     outputs=result_prompt
   )
 
-
-
-
 demo.launch()
 
 ""

input_fields.json

@@ -150,7 +150,7 @@
             "description": "📸 Versatile DSLR Camera"
         },
         {
-            "name": "EOS 5D Mark IV",
+            "name": "GoPro HERO11",
             "description": "✨📷 Action Camera for extreme sports & fish eye effect"
         }
     ],
@@ -227,31 +227,38 @@
     "aspect_ratio_variations": [
         {
             "name": "Square (1:1)",
-            "description": "This is Midjourney's go-to default."
+            "description": "This is Midjourney's go-to default.",
+            "param": "--ar 1:1"
         },
         {
             "name": "Print Lovers (3:2)",
-            "description": "Often chosen for print photography."
+            "description": "Often chosen for print photography.",
+            "param": "--ar 3:2"
         },
         {
             "name": "Retro Vibes (4:3)",
-            "description": "A favorite for older TVs and computer screens."
+            "description": "A favorite for older TVs and computer screens.",
+            "param": "--ar 4:3"
         },
         {
             "name": "Frame Perfect (5:4)",
-            "description": "Widely used for framing and print."
+            "description": "Widely used for framing and print.",
+            "param": "--ar 5:4"
         },
         {
             "name": "Modern Feel (7:4)",
-            "description": "Aligns with HD TV screens and many smartphones."
+            "description": "Aligns with HD TV screens and many smartphones.",
+            "param": "--ar 7:4"
         },
         {
             "name": "Cinema Style (16:9)",
-            "description": "The pick for widescreen displays."
+            "description": "The pick for widescreen displays.",
+            "param": "--ar 16:9"
         },
         {
             "name": "Vertical Shot (9:16)",
-            "description": "Think of it as 16:9 flipped, ideal for portraits."
+            "description": "Think of it as 16:9 flipped, ideal for portraits.",
+            "param": "--ar 9:16"
         }
     ],
     "temperature_variations": [

utils.py

@@ -0,0 +1,96 @@
+import json
+import gradio as gr
+import os
+import langchain_openai
+import langchain_core
+from langchain_openai import ChatOpenAI
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+
+def enhance_subject(subject, details):
+  prompt = ChatPromptTemplate.from_messages([
+      ("system", "Generate a clear and concise subject, then provide additional details using descriptive language. Ensure the response is specific and avoids ambiguity or contradictions. The subject should inspire an engaging photo that tells a story. Remove any unnecessary information and don't add any punctuation at the end of the subject."),
+      ("user", "The main subject is {subject} {details}.")
+    ])
+  output_parser = StrOutputParser()
+  model = ChatOpenAI(model="gpt-3.5-turbo")
+  chain = ( prompt
+    | model
+    | output_parser
+  )
+  result = chain.invoke({"subject": subject, "details": details})
+  return result
+
+def load_input_fields(filepath):
+    """
+    Load the input fields from a JSON file.
+
+    Args:
+    - filepath (str): The path to the JSON file containing the input fields.
+
+    Returns:
+    - dict: A dictionary containing the input fields.
+    """
+
+    with open(filepath, "r") as file:
+        input_fields = json.load(file)
+
+    return input_fields
+
+def create_html_string(input_text, highlight_color = "green", container_style = "border: 2px solid black; padding: 2px; font-size: 16px;" ):
+    """
+    Create a HTML string with specific styles applied to highlighted text within square brackets.
+
+    Args:
+    - input_text (str): The input text with portions to be highlighted within square brackets.
+    - optional: highlight_color (str): Color for the highlighted text (e.g., "green").
+    - optional: container_style (str): Any css for inline styling (e.g,, "border: 2px solid black;")
+
+    Returns:
+    - str: A HTML string with the applied styles.
+    """
+
+    # Replace the highlighted text with HTML span elements for styling
+    highlighted_text = input_text.replace("[", f'<span style="color:{highlight_color}; font-weight: bold;">[').replace("]", "]</span>")
+
+    # Construct the full HTML string with the provided styles
+    html_string = f'<p style="{container_style}">{highlighted_text}</p>'
+
+    return html_string
+
+def extract_names(objects):
+    return [obj['name'] for obj in objects if 'name' in obj]
+
+def clearInput():
+    return ""
+
+def format_to_markdown(objects):
+    # Skip None objects
+    formatted_list = [
+        f"> * **{obj.get('name', 'No Name')}** - {obj.get('description', 'No Description')}"
+        for obj in objects if obj is not None and obj["name"] != "None"
+    ]
+    return '\n'.join(formatted_list)
+
+find_filter_by_name = lambda collection, key: next((filter for filter in collection if filter['name'] == key), None)
+
+def display_info(collection, key):
+  markdown_text = format_to_markdown([find_filter_by_name(collection, key)])
+  return gr.Markdown(markdown_text, visible=True)
+
+def enhance_pipeline(isFrog, subject, details):
+  if isFrog and (subject or details):
+    result = enhance_subject(subject, details)
+    return [gr.Textbox(visible=False), gr.Button(visible=False), gr.TextArea(visible=True, value=result)]
+  elif (subject or details) and not isFrog:
+    return [gr.Textbox(visible=True), gr.Button(visible=True), gr.TextArea(visible=False)]
+  else:
+    return [gr.Textbox(visible=False), gr.Button(visible=False), gr.TextArea(visible=False)]
+
+def authenticate(pwd_input, subject, details):
+  if pwd_input == os.environ.get("MAGIC_WORD"):
+    result = enhance_subject(subject, details)
+    return [gr.TextArea(visible=True, value=result), True, gr.Textbox(visible=False), gr.Button(visible=False)]
+  else:
+    raise gr.Error("You are not from our pond! Use your own LLM to add some juice to your prompt.")
+