diff --git a/main.py b/main.py
index fb9ab949532464a1e3c78aad9e2c186be99e2890..e4bae105f48b06bec3f3cf3a6a7a828d609e5f02 100644
--- a/main.py
+++ b/main.py
@@ -1,24 +1,44 @@
 import streamlit as st
-from streamlit_option_menu import option_menu
+import inscode
 
-# 设置Streamlit应用程序的标题
-st.set_page_config(page_title="app name", layout="wide")
+# 创建页面标题
+st.title("AI文本归纳工具")
 
-menu1="菜单1"
-menu2="菜单2"
+# 定义用户输入框
+input_text = st.text_area("请输入文本内容：", "")
 
-with st.sidebar:
-    menu = option_menu("菜单", [menu1, menu2], 
-        icons=['house', "list-task"], 
-        menu_icon="cast", default_index=0)
+# 定义用户选择框
+option_list = ["提取关键词", "提取摘要", "提取情感"]
+option = st.selectbox("请选择需要提取的内容：", option_list)
 
-def main():
+# 给出示例文本
+st.subheader("示例文本：")
+st.write("汉语是世界上使用人数最多的语言。")
 
-    if menu == menu1:
-        st.subheader(f"{menu1}")
+# 按钮响应函数
+def summarize_text():
+    # 使用 AI 提取关键信息
+    result = inscode.ai("提取关键词", input_text)
+    return result
 
-    if menu == menu2:
-        st.subheader(f"{menu2}")
+def extract_summary():
+    # 使用 AI 提取摘要信息
+    result = inscode.ai("提取摘要", input_text)
+    return result
 
-if __name__ == '__main__':
-    main()
+def extract_sentiment():
+    # 使用 AI 提取情感信息
+    result = inscode.ai("提取情感", input_text)
+    return result
+
+# 根据用户选择调用相应的函数
+if st.button("开始处理"):
+    if option == "提取关键词":
+        result = summarize_text()
+        st.write("关键词：", result)
+    elif option == "提取摘要":
+        result = extract_summary()
+        st.write("摘要：", result)
+    else:
+        result = extract_sentiment()
+        st.write("情感分析：", result)
\ No newline at end of file