BlogGPT/BlogGPT.py at main · vashhdev/BlogGPT · 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
#!/usr/bin/env python
# coding: utf-8

# In[41]:


# from ctypes import alignment
# from tkinter import CENTER
import cohere
co = cohere.Client('vWFpPgxRhWo41o9QhjNiE7VWc3dvulGZwJhawHYk') # This is your trial API key


# In[42]:


def write_post(topic):
    response = co.generate(
      model='command-xlarge-nightly',
      prompt=f'write a blog post on the topic of \"{topic}\"',
      max_tokens=300,
      temperature=0.9,
      k=0,
      p=0.75,
      frequency_penalty=0,
      presence_penalty=0,
      stop_sequences=[],
      return_likelihoods='NONE')
    return response.generations[0].text

def write_poem(topic):
    response = co.generate(
      model='command-xlarge-nightly',
      prompt=f'write a poem on the topic of \"{topic}\"',
      max_tokens=202,
      temperature=1,
      k=0,
      p=0.75,
      frequency_penalty=0,
      presence_penalty=0,
      stop_sequences=[],
      return_likelihoods='NONE')
    return response.generations[0].text

def write_story(topic):
    response = co.generate(
      model='command-xlarge-nightly',
      prompt=f'write a short story on the topic of \"{topic}\"',
      max_tokens=500,
      temperature=0.9,
      k=0,
      p=0.75,
      frequency_penalty=0,
      presence_penalty=0,
      stop_sequences=[],
      return_likelihoods='NONE')
    return response.generations[0].text


# In[44]:


import streamlit as st
# def main():
#     st.sidebar.title("Navigation")
#     selection = st.sidebar.radio("Select a function", ["Write a Blog Post", "Write a Poem", "Write a Short Story"])

#     topic = st.text_input("Enter a topic")
#     if st.button("Generate"):
#         if selection == "Write a Blog Post":
#             result = write_post(topic)
#         elif selection == "Write a Poem":
#             result = write_poem(topic)
#         else:
#             result = write_story(topic)
#         st.success(result)

# if __name__ == '__main__':
#     main()
def main():
    st.sidebar.image("logo.jpg", width= 300)
    st.sidebar.title("What do you want to generate")
    app_mode = st.sidebar.selectbox("Choose one", ["Blog Post", "Poem", "Story"])

    if app_mode == "Blog Post":
        st.title("Blog Post Generator")
        topic = st.text_input("Enter topic:")
        if st.button("Generate Blog Post"):
            blog_post = write_post(topic)
            st.write("Generated Blog Post:")
            st.write(blog_post)

    elif app_mode == "Poem":
        st.title("Poem Generator")
        topic = st.text_input("Enter topic:")
        if st.button("Generate Poem"):
            poem = write_poem(topic)
            st.write("Generated Poem:")
            st.write(poem)

    elif app_mode == "Story":
        st.title("Story Generator")
        topic = st.text_input("Enter topic:")
        if st.button("Generate Story"):
            story = write_story(topic)
            st.write("Generated Story:")
            st.write(story)

if __name__ == '__main__':
    main()