Update app.py

app.py

@@ -128,52 +128,39 @@ Paragraph:
 
 # ===================== 4. Gradio UI（宽文本框 + 全宽布局） =====================
 
-with gr.Blocks(css="""
-.wide_textbox textarea {
-    font-size: 15px !important;
-    line-height: 1.5 !important;
-    width: 100% !important;
-    min-height: 140px !important;
-}
-""") as demo:
-
-    gr.Markdown("# **ChartSmith – AI 论文图表生成助手（支持上传论文 PDF）**")
+with gr.Blocks() as demo:
+    gr.Markdown("# ChartSmith – AI 论文图表生成助手")
 
     with gr.Row():
-        with gr.Column(scale=1):
-            img_in = gr.Image(type="pil", label="上传你的学术图表")
-            paper_file = gr.File(label="上传论文 PDF（可选）", file_types=[".pdf"])
-            language = gr.Radio(["English", "中文"], value="English", label="输出语言")
-            btn = gr.Button("分析并美化图表", variant="primary")
-
-        with gr.Column(scale=1.2):
-            orig_img = gr.Image(label="原始图表")
+        with gr.Column():
+            img_in = gr.Image(type="pil", label="上传你的学术图表（截图也可以）")
 
-            raw_caption_box = gr.Textbox(
-                label="Step 2: 初步自动描述（Vision-LLM）",
-                lines=4,
-                elem_classes=["wide_textbox"]
+            style = gr.Dropdown(
+                ["Formal academic", "Infographic", "Magazine-style"],
+                value="Formal academic",
+                label="重绘风格"
             )
 
-            academic_box = gr.Textbox(
-                label="Step 3: 学术化解释（结合论文上下文）",
-                lines=6,
-                elem_classes=["wide_textbox"]
+            language = gr.Radio(
+                ["English", "中文"],
+                value="English",
+                label="输出语言"
             )
 
-            enhanced_img = gr.Image(label="Step 4: 增强后图示（对比度+亮度）")
-
-            caption_box = gr.Textbox(
-                label="Step 5: 自动生成图注（Caption）",
-                lines=4,
-                elem_classes=["wide_textbox"]
+            paper_file = gr.File(
+                label="上传相关论文（可选，用于更精准解释）",
+                file_types=[".txt", ".pdf", ".md"]
             )
 
-            summary_box = gr.Textbox(
-                label="Step 5: 图表摘要 / 讨论",
-                lines=6,
-                elem_classes=["wide_textbox"]
-            )
+            btn = gr.Button("分析并美化图表")
+
+        with gr.Column():
+            orig_img = gr.Image(label="原始图表")
+            raw_caption_box = gr.Textbox(label="Step 2: 初步自动描述（Vision-LLM）", lines=4)
+            academic_box = gr.Textbox(label="Step 3: 学术化解释（结合论文上下文）", lines=6)
+            enhanced_img = gr.Image(label="Step 4: 强化后的图表（亮度/对比度提升）")
+            caption_box = gr.Textbox(label="Step 5: 自动生成图注（Caption）", lines=4)
+            summary_box = gr.Textbox(label="Step 5: 图表简短摘要 / 讨论", lines=6)
 
     btn.click(
         analyze_and_enhance,