added cohere for better stories

App/Generate/Schema.py

@@ -5,3 +5,4 @@ from pydantic import validator
 
 class GeneratorRequest(BaseModel):
     prompt: str
+    grok: Optional[bool] = True

App/Generate/generatorRoutes.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter, HTTPException, status, BackgroundTasks, UploadFile, Query
 from .Schema import GeneratorRequest
-from .utils.GroqInstruct import chatbot
+from .utils.GroqInstruct import chatbot, VideoOutput
+from .utils.HuggingChat import chatbot as huggingChat
 from .Story.Story import Story
 import asyncio, pprint, json
 from tqdm import tqdm
@@ -18,8 +19,21 @@ async def update_scene(model_scene):
 async def main(request: GeneratorRequest):
     topic = request.prompt
     renderr = RenderVideo()
-    message = chatbot(Prompt.format(topic=topic))
+    if request.grok:
+        message = chatbot(Prompt.format(topic=topic))
 
+    else:
+        message = json.loads(
+            str(
+                huggingChat.query(
+                    Prompt.format(topic=topic)
+                    + f"Match your response to the following schema:  {VideoOutput.model_json_schema()} Make sure to return an instance of the JSON, not the schema itself, and nothing else."
+                )
+            )
+            .split("```json")[1]
+            .split("```")[0]
+            .strip()
+        )
     generated_story = Story.from_dict(message["scenes"])
 
     print("Generated Story ✅")

App/Generate/utils/HuggingChat.py

@@ -20,7 +20,7 @@ cookies = sign.loadCookiesFromDir(cookie_path_dir)
 # Create a ChatBot
 chatbot = hugchat.ChatBot(
     cookies=cookies.get_dict(),
-    system_prompt="You are an amazing youtuber. A true creative master genius",
+    system_prompt="You are  a true creative master genius and a great story teller that keeps the viewer/listener engauged. Make sure that you narrate the sequence of events properly so that the listener can understand. Use smart/insiteful quotes from the book. Don't speak to the viewers just tell the story accurately. Each scene should carry one topic and if the narration is long add more image_prompts, by default a short naration should have  2 image_prompts",
 )
 
 model_index = 0
@@ -31,6 +31,6 @@ if not chatbot.active_model.name == "CohereForAI/c4ai-command-r-plus":
         print(model.name, "switching..")
         if model.name == "CohereForAI/c4ai-command-r-plus":
             model_index = models.index(model)
+            chatbot.switch_llm(model_index)
             break
 print(chatbot.current_conversation.system_prompt)
-chatbot.switch_llm(model_index)

App/Worker.py

@@ -109,9 +109,7 @@ def create_symlink(source_dir, target_dir, symlink_name):
 
 
 def download_with_wget(link, download_dir, filename):
-    print(["aria2c", link, "-d", download_dir, "-o", filename])
     subprocess.run(["aria2c", link, "-d", download_dir, "-o", filename])
-    # os.system(f"aria2c  {link}   -d   {download_dir} -o  {filename}")
 
 
 # @celery.task(name="CopyRemotion")