Update app.py

app.py

@@ -27,7 +27,8 @@ class GradioUI:
         ]
         return random.choice(titles)
 
-    def generate_text(self, prompt):
+    def generate_text(self, title, story):
+        prompt = f"{title}. {story}"
         url = f'https://api-inference.huggingface.co/models/{self.model_id}'
         headers = {
             'Authorization': f'Bearer {self.token}',
@@ -58,11 +59,12 @@ class GradioUI:
         with gr.Blocks(css=self.styles) as demo:
             title = gr.Textbox(placeholder="Title", value=self.random_title())
             random_title_btn = gr.Button("Get Random Title")
-            random_title_btn.click(fn=None, inputs=None, outputs=[title], _js=f"return ['{self.random_title()}'];")
+            random_title_btn.click(fn=self.random_title, inputs=None, outputs=[title])
             editor = gr.Textbox(placeholder="Write your story here.", lines=32, max_lines=32, elem_classes=['no-label', 'small-big-textarea'])
             gen_btn = gr.Button("Generate Text")
             chatbot = gr.Chatbot([])
-            chat_txt = gr.Textbox(placeholder="Chat here", elem_classes=['no-label'])
+            
+            gen_btn.click(fn=self.generate_text, inputs=[title, editor], outputs=[chatbot])
             
             # More Gradio elements and logic can be added here.