Update app.py

app.py

@@ -6,15 +6,16 @@ from crewai import Crew, Process
 from agents import get_agents_and_tasks
 
 
-def generate_video(topic, grow_api_key, stability_ai_api_key,openai_api_key):
+def generate_video(topic, stability_ai_api_key):
     # os.environ['GROQ_API_KEY'] = grow_api_key
     os.environ['STABILITY_AI_API_KEY'] = stability_ai_api_key
-    os.environ['OPENAI_API_KEY'] = openai_api_key
+    os.environ['OPENAI_API_KEY'] = 'sk-proj-0zw99wkFRW5V3tc6xZONT3BlbkFJqNbENviPvW4T3YyiktGL'
+    grow_api_key = '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'
     agents, tasks = get_agents_and_tasks(grow_api_key)
 
     crew = Crew(
     # agents=[script_agent, image_descriptive_agent, img_speech_generating_agent, editor],
-    # tasks=[content_generation_task, story_writing_task, img_text_task,speech_generation_task,make_video_task],
+    # tasks=[content_generation_task, story_writing_task, img_text_task, img_generation_task,speech_generation_task,make_video_task],
     agents = agents, 
     tasks = tasks,
     process = Process.sequential,
@@ -25,14 +26,39 @@ def generate_video(topic, grow_api_key, stability_ai_api_key,openai_api_key):
     crew.kickoff(inputs={'topic': topic})
     return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'outputs/final_video/final_video.mp4')
 
-app = gr.Interface(
-    fn=generate_video,
-    inputs=['text', 'text', 'text', 'text'],
-    # outputs=gr.Video(value=os.path.join('outputs/final_video/video.mp4'),label="Generated Video", width=720/2, height=1280/2),
-    outputs = gr.Video(format='mp4',label="Generated Video", width=720/2, height=1280/2),
-    title="ShortsIn",
-    description="Shorts generator"
-)
+# app = gr.Interface(
+#     fn=generate_video,
+#     inputs=['text', 'text'],
+#     # outputs=gr.Video(value=os.path.join('outputs/final_video/video.mp4'),label="Generated Video", width=720/2, height=1280/2),
+#     outputs = gr.Video(format='mp4',label="Generated Video", width=720/2, height=1280/2),
+#     title="ShortsIn",
+#     description="Shorts generator"
+# )
+intro = """
+<div style="text-align:center">
+<h1 style="font-weight: 1700; text-align: center; margin-bottom: 7px;">
+   ShortsIn
+</h1>
+<div style="display:flex; justify-content: center;margin-top: 0.5em; color: white;">
+convert text to Youtube Shorts
+</div>
+</div>
+"""
+
+with gr.Blocks(title='ShortsIn') as app:
+
+  gr.HTML(intro)
+
+  with gr.Row():
+    with gr.Column():
+      inp = gr.Textbox(label='Enter title here')
+      api = gr.Textbox(label='Enter Stibility ai API key here')
+      btn = gr.Button('Generate', size='lg')
+    with gr.Column():
+      out = gr.Video(label='', height=720, width=405)
+      
+
+  btn.click(fn=generate_video, inputs=[inp, api], outputs=out)
 
 app.launch()
 #os.path.dirname(os.path.abspath(__file__))