Update app.py

app.py

@@ -4,20 +4,24 @@ import requests
 from PIL import Image
 import io
 
-def generate_text(prompt,maxi,cohere_api_key):
+def generate_text(prompt, maxi, cohere_api_key):
     co = cohere.Client(cohere_api_key)
 
-    response = co.generate(prompt=prompt,
-                            temperature=0,
-                            max_tokens=maxi)
-    return response[0]
+    response = co.generate(
+        model='xlarge',  # Specify the model you are using
+        prompt=prompt,
+        temperature=0.7,
+        max_tokens=maxi,
+        stop_sequences=["--END--"]
+    )
+    return response.generations[0].text
 
-def generate_image(prompt,gradio_api_key):
+def generate_image(prompt, gradio_api_key):
     r = requests.post('https://clipdrop-api.co/text-to-image/v1',
-        files={
-            'prompt': (None, prompt, 'text/plain')
-        },
-        headers={'x-api-key': gradio_api_key}
+                      files={
+                          'prompt': (None, prompt, 'text/plain')
+                      },
+                      headers={'x-api-key': gradio_api_key}
     )
 
     if r.ok:
@@ -25,22 +29,29 @@ def generate_image(prompt,gradio_api_key):
         return images
     else:
         raise ValueError("Failed to generate image")
-    
-def text_and_image_generator(prompt,maxi,cohere_api_key,gradio_api_key):
-    cohere_api_key = "3kUv9DkIN6vtDUhUXFnYyk3cqy53VWuDfC6g3mp4"
-    gradio_api_key = "e35a803cffac3d96f1c364a1bb8ddf00fd702be129881b1acb36f963c0ba86489cad39a6cb6f15f7cbfe919a3c4eb46e"
-    text = generate_text(f"Generate a story with {prompt}. Conclude the generated story properly.",maxi,cohere_api_key)
-    image = generate_image(text,gradio_api_key)
-    title = generate_text(f"title for the story {prompt} within 5 words",5,cohere_api_key)
 
-    return title,text,image
+def text_and_image_generator(prompt, maxi, cohere_api_key, gradio_api_key):
+    text = generate_text(f"Generate a story with {prompt}. Conclude the generated story properly.", maxi, cohere_api_key)
+    image = generate_image(text, gradio_api_key)
+    title = generate_text(f"title for the story {prompt} within 5 words", 5, cohere_api_key)
+
+    return title, text, image
 
 app = gr.Interface(
-    title="Story and Image Generator",
     fn=text_and_image_generator,
-    inputs = [gr.inputs.Textbox(label="Enter your prompt to generate a story"),
-    gr.inputs.Slider(320,1000,label="Story length")],
-    outputs= [gr.outputs.Textbox(label="Story title"),gr.outputs.Textbox(label="Story"),gr.outputs.Image(type="pil",label="Image based on the Generated story")],theme="dark"
+    inputs=[
+        gr.Textbox(label="Enter your prompt to generate a story", lines=2, placeholder="Type your prompt here..."),
+        gr.Slider(320, 1000, step=1, label="Story length"),
+        gr.Textbox(label="Cohere API Key", type="password"),
+        gr.Textbox(label="Gradio API Key", type="password")
+    ],
+    outputs=[
+        gr.Textbox(label="Story title"),
+        gr.Textbox(label="Story"),
+        gr.Image(type="pil", label="Image based on the Generated story")
+    ],
+    title="Story and Image Generator",
+    theme="dark"
 )
 
-app.launch()
+app.launch()