Update app.py

app.py

@@ -1,7 +1,36 @@
 import gradio as gr
 
-def image_classifier(inp):
-    return {'cat': 0.3, 'dog': 0.7}
+def generate_text(title, context):
+    return f"Title:{title}\nContext:{context}\n..."
 
-demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
-demo.launch()
+def generate_mutimodal(title, context, img):
+    return f"Title:{title}\nContext:{context}\n...{img}"
+
+with gr.Blocks() as demo:
+    # 顶部文字
+    gr.Markdown("# 多模态可控XX生成")
+
+    # 多个tab
+    with gr.Tabs():
+
+        with gr.TabItem("新闻输入"):
+            title = gr.Textbox(label="标题", lines=1, placeholder="请输入标题")
+            context = gr.Textbox(label="正文", lines=2, placeholder="请输入正文")
+            text_button = gr.Button("生成")
+            
+            text_output = gr.Textbox(label="生成内容", lines=5, placeholder="生成列表")
+
+
+        with gr.TabItem("图文输入"):
+            title = gr.Textbox(label="标题", lines=1, placeholder="请输入标题")
+            context = gr.Textbox(label="正文", lines=2, placeholder="请输入正文")
+            img = gr.Image(shape=(200, 200),  label="请上传图片")
+            img_button = gr.Button("生成")
+
+            img_output = gr.Textbox(label="生成内容", lines=5, placeholder="生成列表")
+
+
+    text_button.click(fn=generate_text, inputs=[title,context], outputs=text_output)
+    img_button.click(fn=generate_mutimodal, inputs=[title,context,img], outputs=img_output)
+
+demo.launch()