commit

app.py

@@ -0,0 +1,87 @@
+import os
+import streamlit as st
+from langchain.prompts import PromptTemplate
+from dotenv import load_dotenv
+import google.generativeai as genai
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain.prompts import PromptTemplate
+# Load environment variables from .env file
+load_dotenv()
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+model_gemini = ChatGoogleGenerativeAI(model="gemini-pro")
+
+
+def model(genre,no_of_characters,location,characters_prompt,hook_prompt,climax_prompt):
+    # here we are creating a template for the prompt
+    main_c = """
+    Crafting a compelling short film necessitates a concise and straightforward premise, a key ingredient for success when storytelling within the constraints of a limited timeframe. The best short films often feature a main character with a specific goal or a tight deadline, such as two friends embarking on a bike ride, a woman seeking new friendship, or a grieving son delivering a eulogy. With time being a precious commodity, short films cannot afford to delve extensively into character backstories. Instead, filmmakers must discern the essential information required to engage the audience in the hero's journey. 
+    Details like an ex-husband may prove extraneous and dilute the impact of the narrative.Moreover, judiciously selecting the number of characters is pivotal. 
+    Each character, whether a protagonist, supporting role, or minor part, should serve a purpose directly tied to the central goal. 
+    If a character's exclusion doesn't impede the main character's progression, 
+    it might be prudent to omit them. Many successful short films unfold within a singular location, not only to streamline the story but also to mitigate budgetary constraints, 
+    particularly for independent filmmakers. "Sam Did It," a renowned 10-minute short set entirely in a morgue operating room, exemplifies how a confined space can intensify the storytelling experience.
+     A useful exercise during brainstorming involves formulating a premise and identifying a single location, prompting creators to contemplate whether the entire narrative can unfold within those confines. 
+    This approach not only sharpens the focus but also aligns with budget considerations for those contemplating a self-produced project.
+    
+    """
+
+    template = main_c +"""
+    write a short film story in {genre} genre with {no_of_characters} characters in {location} location
+    Consider being economical with characters and backstory There are {characters_prompt} characters and their behaviours. This is the hook {hook_prompt} 
+    and this is the climax {climax_prompt}
+    """
+
+    # here we are creating a prompt using the template and the input variables
+    prompt = PromptTemplate(input_variables=["genre","no_of_characters","location","characters_prompt","hook_prompt","climax_prompt"],template=template)
+
+    # here we are generating the blog
+    response = model_gemini.invoke(prompt.format(genre=genre,no_of_characters=no_of_characters,location=location,characters_prompt=characters_prompt,hook_prompt=hook_prompt,climax_prompt=climax_prompt))
+    print(response)
+    return response.content
+
+def short_film_story_generator(genre):
+    st.title("Short Film Story Generator")
+
+    # Prompt for being economical with characters and backstory
+    no_of_characters = st.text_input("How many character do you want? (leave empty for default of 4 characters):")
+    if not no_of_characters:
+        # Default characters if the user leaves it empty
+        no_of_characters = "The story follows four characters who..."
+    # Prompt for minimal locations
+    location = st.text_input("Think about keeping the locations to a minimum. Specify a key location for your short film (leave empty for random location):")
+    if not location:
+        # Default location if the user leaves it empty
+        location = "You can use any location you want"
+
+    # Prompt for interesting characters
+    characters_prompt = st.text_input("Describe interesting traits or quirks of the main characters:")
+    if not characters_prompt:
+        characters_prompt = "You can use any interesting traits or quirks you want"
+
+    # Prompt for the hook
+    hook_prompt = st.text_input("Create a hook for your short film:")
+    if not hook_prompt:
+        # Default hook if the user leaves it empty
+        hook_prompt = "You can use any hook you want"
+
+
+    # Prompt for the climax
+    climax_prompt = st.text_input("Develop a great climax for your short film:")
+    if not climax_prompt:
+        climax_prompt = "You can use climax you want but should match with the genre"
+
+
+    if st.button("Generate Story"):
+        response = model(genre,no_of_characters,location,characters_prompt,hook_prompt,climax_prompt)
+
+        # Extract the AI's response and display it
+        ai_response = response
+        st.write(f"{ai_response}")
+
+if __name__ == "__main__":
+    genres = ["Sci-Fi", "Drama", "Comedy", "Thriller", "Mystery", "Fantasy", "Horror", "Romance", "Action", "Adventure"]
+    
+    # Create a dropdown menu for genres
+    user_genre = st.sidebar.selectbox("Select the genre:", genres, index=genres.index("Sci-Fi"))
+
+    short_film_story_generator(user_genre)

requirements.txt

@@ -0,0 +1,4 @@
+google-generativeai
+streamlit
+python-dotenv
+langchain