gen_ai_hackathon/main.py at master · gor2000/gen_ai_hackathon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
# main.py
from pathlib import Path

import streamlit as st
import requests
import os
import time
import development as dev
from io import StringIO


def main(myclient):
    st.set_page_config(page_title="AI Whisperers", page_icon="📽️")
    st.title("AI Whisperers")
    user_input_link = st.text_input("Enter a Link from a Newspaper:")
    # Display the link after it's entered
    if user_input_link:
        article = dev.get_article(user_input_link)
        st.success(f"Link successfully added!")
        summary_1 = dev.create_summary(myclient, article)
        summary_2 = dev.create_summary(myclient, article)
        summary_3 = dev.create_summary(myclient, article)

        summaries_dict = {
            'summary_1': summary_1,
            'summary_2': summary_2,
            'summary_3': summary_3,
        }
        st.markdown("## Summaries")

        # Display summaries using expander
        for i in range(1, 4):
            summary_key = f'summary_{i}'
            with st.expander(f"Summary {i}", expanded=True):
                st.write(summaries_dict[summary_key])

        # Allow the user to select their favorite summary
        favorite_summary = st.selectbox(
            "Select your favorite summary:",
            ('Summary 1', 'Summary 2', 'Summary 3')
        )

        if favorite_summary:
            st.write("You selected:", favorite_summary)
            if favorite_summary == 'Summary 1':
                sum_fav = summaries_dict['summary_1']
            elif favorite_summary == 'Summary 2':
                sum_fav = summaries_dict['summary_2']
            else:
                sum_fav = summaries_dict['summary_3']

        st.markdown("## Upload your video")
        uploaded_file = st.file_uploader("Upload a video", type=['mp4', 'mp3'])
        # Directory where the file will be temporarily saved
        temp_dir = "tempDir"

        # Ensure temp_dir exists
        if not os.path.exists(temp_dir):
            os.makedirs(temp_dir)

        if uploaded_file is not None:
            # Path to save the uploaded file
            temp_file_path = os.path.join(temp_dir, uploaded_file.name)

            # Write the file to the temporary directory
            with open(temp_file_path, "wb") as f:
                f.write(uploaded_file.getbuffer())
            st.success("File saved!")
            st.video(temp_file_path)

            with open(temp_file_path, "rb") as audio_file:
                transcript = myclient.audio.transcriptions.create(
                    model="whisper-1",
                    file=audio_file
                )

                with st.expander('Video Transcript'):
                    st.write(transcript.text)

                seconds = st.selectbox("Select the video duration in seconds:", ('30', '60'))
                if seconds == '30':
                    s = 30
                else:
                    s = 60
                if st.button('Generate Script', type='primary'):
                    script_1 = dev.generate_script(transcript.text, article, s, myclient)
                    script_2 = dev.generate_script(transcript.text, article, s, myclient)
                    script_3 = dev.generate_script(transcript.text, article, s, myclient)

                    script_dicc = {
                        'script_1': script_1,
                        'script_2': script_2,
                        'script_3': script_3,
                    }
                    print(script_dicc)
                    st.markdown("## Scripts")
                    for i in range(1, 4):
                        script_key = f'script_{i}'
                        with st.expander(f"Script {i}", expanded=True):
                            st.write(script_dicc[script_key])

                    favorite_script = st.selectbox(
                        "Select your favorite script:",
                        ('Script 1', 'Script 2', 'Script 3')
                    )

                    if favorite_script:
                        st.write("You selected:", favorite_script)
                        if favorite_script == 'Script 1':
                            script_fav = script_dicc['script_1']
                        elif favorite_summary == 'Script 2':
                            script_fav = script_dicc['script_2']
                        else:
                            script_fav = script_dicc['script_3']

                    # Button to convert text to speech

                    if script_fav:
                        # Call your text_to_audio function
                        dev.text_to_audio(myclient, script_fav)

                        # Assuming the function saves the audio file in the same directory as the script
                        audio_file = Path(__file__).parent / "speech.mp3"

                        # Display audio player
                        if audio_file.is_file():
                            audio_bytes = open(audio_file, "rb").read()
                            st.audio(audio_bytes, format="audio/mp3")
                        else:
                            st.error("Error: Audio file not found.")


if __name__ == "__main__":
    my_client = dev.init_openai()
    main(my_client)