Upload 3 files

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
 from streamlit_option_menu import option_menu
-from generate_plot import generate_plot, generate_storybook, generate_book_cover
-from generate_horror_plot import generate_horror_plot, generate_horror_storybook
+from generate_plot import generate_plot, generate_storybook, generate_book_cover, set_openai_api_key
+from generate_horror_plot import generate_horror_plot, generate_horror_storybook, set_openai_api_key
 from markup import app_intro, how_use_intro
+from g4f import Provider
 
 
 if "generated_code" not in st.session_state:
@@ -31,8 +32,6 @@ def tab2():
     st.markdown("Create an enchanting children's book with custom characters and illustrations!")
     st.write("Follow the magical instructions below:")
 
-    #openai_api_key = st.text_input("🔑 Enter your OpenAI API key:", type='password')
-
     col1, col2 = st.columns(2)
     with col1:
         st.subheader("Step 1: Enter the Age 🧒")
@@ -54,13 +53,30 @@ def tab2():
         selected_style = st.selectbox("Select Art Style:", art_styles)
 
     result = None
-
-    if st.button("Generate Plot"):
-
-        with st.spinner('Generating plot...'):
-            result = generate_plot(number_of_pages, character_names, age, selected_style)
-        st.code(result)
-        st.session_state.generated_code = result
+    st.subheader("Step 5: Choose API 🔌")
+    st.write("Select between G4F free providers or OpenAI. Please note that some G4F providers may not work all time")
+    provider = st.selectbox("Select API:", ["G4F", "OpenAI"])
+    
+    if provider == "OpenAI":
+        openai_api_key = st.text_input("🔑 Enter your OpenAI API key:", type='password')
+        if st.button("Generate Plot") and openai_api_key:
+            set_openai_api_key(openai_api_key)
+            with st.spinner('Generating plot...'):
+                result = generate_plot(number_of_pages, character_names, age, selected_style, provider="OpenAI")
+            st.code(result)
+            st.session_state.generated_code = result
+        elif not openai_api_key:
+            st.warning("Please enter your OpenAI API key.")
+    elif provider == "G4F":
+        selected_provider = st.selectbox("Select G4F Provider:", [
+            "Ails", "You", "Aichat", "Bard",
+            "Forefront", "DeepAi", "GetGpt"
+        ])
+        if st.button("Generate Plot"):
+            with st.spinner('Generating plot...'):
+                result = generate_plot(number_of_pages, character_names, age, selected_style, provider="G4F", selected_provider=selected_provider)
+            st.code(result)
+            st.session_state.generated_code = result
 
 
 
@@ -93,7 +109,7 @@ def tab3():
 
         st.subheader("Step 1: Choose the Number of Pages 📖")
         st.write("Select the number of pages for your bone-chilling horror story (between 2 and 7).")
-        number_of_pages = st.number_input("Enter the Number of Pages:",  min_value=2, max_value=7, value=3)
+        number_of_pages = st.number_input("Enter the Number of Pages:",  min_value=2, max_value=7, value=2)
 
     with col2:
 
@@ -103,13 +119,31 @@ def tab3():
         selected_style = st.selectbox("Select Art Style:", art_styles)
 
     result = None
-
-    if st.button("Generate Plot"):
-
-        with st.spinner('Generating plot...'):
-            result = generate_horror_plot(number_of_pages, selected_style)
-        st.code(result)
-        st.session_state.generated_horror_code = result
+    st.subheader("Step 3: Choose API 🔌")
+    st.write("Select between G4F free providers or OpenAI. Please note that some G4F providers may not work all time")
+    provider = st.selectbox("Select API:", ["G4F", "OpenAI"])
+
+    if provider == "OpenAI":
+        openai_api_key = st.text_input("🔑 Enter your OpenAI API key:", type='password')
+        if st.button("Generate Plot") and openai_api_key:
+            set_openai_api_key(openai_api_key)
+            with st.spinner('Generating plot...'):
+                result = generate_horror_plot(number_of_pages, selected_style, provider="OpenAI")
+            st.code(result)
+            st.session_state.generated_code = result
+        elif not openai_api_key:
+            st.warning("Please enter your OpenAI API key.")
+    elif provider == "G4F":
+        selected_provider = st.selectbox("Select G4F Provider:", [
+            "Ails", "You", "Aichat", "Bard",
+            "Forefront", "DeepAi", "GetGpt"
+        ])
+        if st.button("Generate Plot"):
+
+            with st.spinner('Generating plot...'):
+                result = generate_horror_plot(number_of_pages, selected_style, provider="G4F", selected_provider=selected_provider)
+            st.code(result)
+            st.session_state.generated_horror_code = result
 
     if st.button("Generate Horror Storybook"):
         if not st.session_state.generated_horror_code:

generate_horror_plot.py

@@ -29,8 +29,8 @@ Answer:"""
 prompt = PromptTemplate(template=template, input_variables=["number_of_pages"])
 
 def query(payload):
-    #API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
-    API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/all-526-animated"
+    API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
+    #API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/all-526-animated"
     headers = {"Authorization": "Bearer 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"}
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
@@ -41,12 +41,51 @@ def query_alt(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
-def generate_horror_plot(number_of_pages, selected_style):
-    llm = G4FLLM(
-        model=Model.gpt_35_turbo,
-        provider=Provider.GetGpt,
-    )
-    
+def generate_horror_plot(number_of_pages, selected_style, provider, selected_provider=None):
+    if provider == "OpenAI":
+        llm = OpenAI(temperature=0)
+    elif provider == "G4F":
+        if selected_provider == "Ails":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Ails,
+            )
+        elif selected_provider == "You":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.You,
+            )
+        elif selected_provider == "GetGpt":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.GetGpt,
+            )
+        elif selected_provider == "DeepAi":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.DeepAi,
+            )
+        elif selected_provider == "Forefront":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Forefront,
+            )
+        elif selected_provider == "Aichat":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Aichat,
+            )
+        elif selected_provider == "Bard":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Bard,
+            )
+        # Add other providers here
+        else:
+            raise ValueError("Invalid G4F provider selected.")
+    else:
+        raise ValueError("Invalid provider selected.")
+
     llm_chain = LLMChain(prompt=prompt, llm=llm)
     response = llm_chain.run(number_of_pages=number_of_pages)
     pages = response.split("<Text>")

generate_plot.py

@@ -42,12 +42,51 @@ def query_alt(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
-def generate_plot(number_of_pages, character_names, age, selected_style):
-    llm = G4FLLM(
-        model=Model.gpt_35_turbo,
-        provider=Provider.GetGpt,
-    )
-    
+def generate_plot(number_of_pages, character_names, age, selected_style, provider, selected_provider=None):
+    if provider == "OpenAI":
+        llm = OpenAI(temperature=0)
+    elif provider == "G4F":
+        if selected_provider == "Ails":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Ails,
+            )
+        elif selected_provider == "You":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.You,
+            )
+        elif selected_provider == "GetGpt":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.GetGpt,
+            )
+        elif selected_provider == "DeepAi":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.DeepAi,
+            )
+        elif selected_provider == "Forefront":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Forefront,
+            )
+        elif selected_provider == "Aichat":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Aichat,
+            )
+        elif selected_provider == "Bard":
+            llm = G4FLLM(
+                model=Model.gpt_35_turbo,
+                provider=Provider.Bard,
+            )
+        # Add other providers here
+        else:
+            raise ValueError("Invalid G4F provider selected.")
+    else:
+        raise ValueError("Invalid provider selected.")
+
     llm_chain = LLMChain(prompt=prompt, llm=llm)
     response = llm_chain.run(number_of_pages=number_of_pages, character_names=character_names, age=age)
     pages = response.split("<Text>")