Update app.py

app.py

@@ -1,15 +1,11 @@
-# Import necessary libraries
+import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from pptx import Presentation
 from pptx.util import Inches
-import streamlit as st
 import subprocess
 import os
-try:
-    from transformers import AutoTokenizer, AutoModelForCausalLM
-except ImportError:
-    print("Transformers library is not installed. Please install it using pip install transformers.")
-# Content Generation Agent
+
+# Content Generation Function
 def generate_content(prompt):
     tokenizer = AutoTokenizer.from_pretrained("gpt2")
     model = AutoModelForCausalLM.from_pretrained("gpt2")
@@ -18,64 +14,49 @@ def generate_content(prompt):
     text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return text
 
-# Slide Design Agent
+# Slide Design Function
 def create_presentation(content_dict, output_file):
     prs = Presentation()
-    # Assume template has a title slide and content slides
-    # Create title slide
-    title_slide_layout = prs.slide_layouts[0]
-    slide = prs.slides.add_slide(title_slide_layout)
-    slide.shapes.title.text = content_dict['title']
-    slide.placeholders[1].text = content_dict['subtitle']
-    # Create content slides
-    for content in content_dict['slides']:
-        content_slide_layout = prs.slide_layouts[1]
-        slide = prs.slides.add_slide(content_slide_layout)
-        slide.shapes.title.text = content['title']
-        slide.placeholders[1].text = content['content']
+    # Create slides based on content_dict
+    # ...
     prs.save(output_file)
 
-# Output Conversion Agent
+# Output Conversion Function
 def convert_to_pdf(pptx_file, pdf_file):
     subprocess.run(['soffice', '--headless', '--convert-to', 'pdf', pptx_file, '--outdir', os.path.dirname(pdf_file)])
 
-# UI Agent
-def main():
-    st.title("Presentation Generator")
-    title = st.text_input("Enter presentation title:")
-    subtitle = st.text_input("Enter subtitle:")
-    num_slides = st.number_input("Number of slides", min_value=1, max_value=10, value=1)
+# Main Function
+def main(title, subtitle, num_slides, slide_prompts):
     slides = []
-    for i in range(num_slides):
-        slide_title = st.text_input(f"Slide {i+1} title:")
-        slide_prompt = st.text_input(f"Slide {i+1} prompt:")
-        if st.button(f"Generate Content for Slide {i+1}"):
-            content = generate_content(slide_prompt)
-            slides.append({'title': slide_title, 'content': content})
-    if st.button("Generate Presentation"):
-        content_dict = {
-            'title': title,
-            'subtitle': subtitle,
-            'slides': slides
-        }
-        pptx_file = "output.pptx"
-        create_presentation(content_dict, pptx_file)
-        pdf_file = "output.pdf"
-        convert_to_pdf(pptx_file, pdf_file)
-        st.success("Presentation generated successfully!")
-        # Provide download links
-        with open(pptx_file, "rb") as file:
-            btn = st.download_button(
-                label="Download PPTX",
-                data=file,
-                file_name="presentation.pptx"
-            )
-        with open(pdf_file, "rb") as file:
-            btn = st.download_button(
-                label="Download PDF",
-                data=file,
-                file_name="presentation.pdf"
-            )
+    for prompt in slide_prompts:
+        content = generate_content(prompt)
+        slides.append({'title': content, 'content': content})
+    content_dict = {
+        'title': title,
+        'subtitle': subtitle,
+        'slides': slides
+    }
+    pptx_file = "output.pptx"
+    create_presentation(content_dict, pptx_file)
+    pdf_file = "output.pdf"
+    convert_to_pdf(pptx_file, pdf_file)
+    return pptx_file, pdf_file
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Presentation Generator")
+    title = gr.Textbox(label="Presentation Title")
+    subtitle = gr.Textbox(label="Subtitle")
+    num_slides = gr.Number(label="Number of Slides", value=1)
+    slide_prompts = gr.Textbox(label="Slide Prompts (one per line)", lines=5)
+    generate_button = gr.Button("Generate Presentation")
+    output_pptx = gr.File(label="Download PPTX")
+    output_pdf = gr.File(label="Download PDF")
+
+    generate_button.click(
+        main,
+        inputs=[title, subtitle, num_slides, slide_prompts],
+        outputs=[output_pptx, output_pdf]
+    )
 
-if __name__ == "__main__":
-    main()
+demo.launch()