improved image prompt + tuned llama model

app.py

@@ -128,42 +128,22 @@ def __getRawImagePromptDetails(prompt: str, response: str) -> Tuple[str, str, st
             "Painting your character ...",
         )
 
-        '''
-        Style: Fantastical, in a storybook, surreal, bokeh
-        '''
-
-        '''
-        Mood: ethereal lighting that emphasizes the fantastical nature of the scene.
-
-        storybook style
-
-        4d model, unreal engine
-
-        Alejandro Bursido
-
-        vintage, nostalgic
-
-        Dreamlike, Mystical, Fantastical, Charming
-        '''
-
     if __matchingKeywordsCount(
-        ["tier 2", "tier-2"],
+        [C.BOOKING_LINK],
         cleanedResponse
     ) > 0:
-        possibleStoryEndIdx = [response.find("tier 2"), response.find("tier-2")]
-        storyEndIdx = max(possibleStoryEndIdx)
-        relevantResponse = response[:storyEndIdx]
-        U.pprint(f"{relevantResponse=}")
+        relevantResponse = f"""
+        {st.session_state.chatHistory[-1].get("content")}
+
+        {response}
+        """
         return (
-            "Extract the story plot from this text:\n{response}",
+            f"Extract the story from this text:\n{relevantResponse}",
             """
             Style: In a storybook, surreal
             """,
             "Imagining your scene (beta) ...",
         )
-        """
-        photo of a scene from this text: {relevantResponse}.
-        """
 
     return (None, None, None)
 
@@ -189,7 +169,13 @@ def __getImagePromptDetails(prompt: str, response: str):
             "model": model,
             "messages": [{
                 "role": "user",
-                "content": f"{promptPrefix} create a prompt for image generation (limit to less than 500 words)\n\n{imagePrompt}"
+                "content": f"""
+{promptPrefix} create a prompt for image generation (limit to less than 500 words)
+
+{imagePrompt}
+
+Return only the final Image Generation Prompt, and nothing else
+"""
             }],
             "temperature": 1,
             "max_tokens": 2000
@@ -243,7 +229,7 @@ def predict():
             with client.messages.stream(
                 model=MODEL,
                 messages=messagesFormatted,
-                temperature=0.7,
+                temperature=0.8,
                 system=C.SYSTEM_MSG,
                 max_tokens=4000,
             ) as stream:
@@ -259,7 +245,7 @@ def predict():
             response = client.chat.completions.create(
                 model=MODEL,
                 messages=messagesFormatted,
-                temperature=0.8,
+                temperature=1,
                 max_tokens=4000,
                 stream=True
             )

constants.py

@@ -1,5 +1,6 @@
 JSON_SEPARATOR = ">>>>"
 EXCEPTION_KEYWORD = "<<EXCEPTION>>"
+BOOKING_LINK = "https://calendly.com"
 
 SYSTEM_MSG = f"""
 => Context:
@@ -104,7 +105,7 @@ If the user has any suggestions, incorporate them and then show the story again.
 
 ## Tier 2: Story Enhancement
 #### After initial story creation, you offer congratulations on completing the first draft and gives 2 options:
-Option 1 - Provides option for one-on-one sessions with expert storytelling coaches - the booking can be done that at https://calendly.com/
+Option 1 - Provides option for one-on-one sessions with expert storytelling coaches - the booking can be done that at {BOOKING_LINK}
 Option 2 - Provides further options for introducing users to more sophisticated narratives.
 
 #### If Option 2 chosen, show these options with simple explanation and chose one.

tools/webScraper.py

@@ -4,13 +4,13 @@ import requests
 
 
 def scrapeGoogleSearch(query):
-    result = []
+    finalResponse = []
 
     searchUrl = f"https://www.google.com/search?q={query}"
     response = requests.get(searchUrl)
     if response.status_code == 200:
         soup = BeautifulSoup(response.text, 'html.parser')
-        with open('soupDump.html', 'w', encoding='utf-8') as file:
+        with open('soup_dump.html', 'w', encoding='utf-8') as file:
             file.write(soup.prettify())
 
         results = soup.find('body')
@@ -24,11 +24,14 @@ def scrapeGoogleSearch(query):
             citationDate = citationDateDiv.text if citationDateDiv else ""
             answerText = answerDiv.text.replace(citationDate, '').strip()
             citationText = f"Citation Date: {citationDate}" if citationDate else ""
-            result.append(f"====\n{answerText}\n{citationText}\n====\n\n")
+            finalResponse.append(f"Verified Answer:\n====\n{answerText}\n{citationText}\n====\n\n")
 
         results = mainDiv.select('div.egMi0.kCrYT')
         resultsDesc = mainDiv.select('div.BNeawe.s3v9rd.AP7Wnd .BNeawe.s3v9rd.AP7Wnd:last-child')
 
+        if results:
+            finalResponse.append("Search Results:\n====\n")
+
         for (i, result) in enumerate(results[:10]):
             title = result.find('h3').text
             link = result.find('a')['href']
@@ -36,10 +39,10 @@ def scrapeGoogleSearch(query):
             urlParams = parse_qs(parsedUrl.query)
             link = urlParams.get('q', [None])[0]
             desc = resultsDesc[i].text
-            result.append(f"Title: {title}")
-            result.append(f"Description: {desc}")
-            result.append(f"Link: {link}\n")
+            finalResponse.append(f"Title: {title}")
+            finalResponse.append(f"Description: {desc}")
+            finalResponse.append(f"Link: {link}\n")
     else:
         print("Failed to retrieve search results.")
 
-    return "".join(result)
+    return "\n".join(finalResponse)