Update app.py

app.py

@@ -87,7 +87,7 @@ def add_contribution(contributor, content):
 
 def get_similar_parts(query, num_results=5):
     sim = story_table.content.similarity(query)
-    results = story_table.order_by(sim, asc=False).limit(num_results).select(story_table.content, story_table.contributor, sim=sim).collect()
+    results = story_table.order_by(sim, asc=False).limit(num_results).select(story_table.content, story_table.contributor).collect()
     return results.to_pandas()
 
 def generate_next_part():
@@ -98,7 +98,6 @@ def summarize_story():
     summary = story_table.select(summary=story_table.summary_response.choices[0].message.content).tail(1)['summary'][0]
     return summary
 
-# Gradio interface
 with gr.Blocks(theme=gr.themes.Base()) as demo:
     gr.HTML(
     """
@@ -122,31 +121,28 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
         with gr.TabItem("Contribute"):
             with gr.Row():
                 with gr.Column(scale=2):
-                    contributor = gr.Textbox(label="Your Name")
+                    contributor = gr.Textbox(label="Your Nickname")
                     content = gr.Textbox(label="Your Contribution", lines=5)
                     submit_btn = gr.Button("Submit Contribution", variant="primary")
-                with gr.Column(scale=3):
-                    status = gr.Textbox(label="Status")
-                    current_story = gr.Textbox(label="Current Story", lines=10, interactive=False)
+            
+            with gr.Row():
+                with gr.Column():
+                    generate_btn = gr.Button("Generate Next Part", variant="primary")
+                    generated_part = gr.Textbox(label="Generated Continuation", lines=5)
+                with gr.Column():
+                    summarize_btn = gr.Button("Summarize Story", variant="primary")
+                    summary = gr.Textbox(label="Story Summary", lines=5)
         
-        with gr.TabItem("Search & Generate"):
+        with gr.TabItem("Search Past Contributions"):
             with gr.Row():
                 with gr.Column():
                     search_query = gr.Textbox(label="Search Current Contributions")
                     num_results = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Number of Results")
                     search_btn = gr.Button("Search", variant="secondary")
                     search_results = gr.Dataframe(
-                        headers=["Content", "Contributor", "Similarity"],
+                        headers=["Content", "Contributor"],
                         label="Similar Parts"
                     )
-                
-                with gr.Column():
-                    generate_btn = gr.Button("Generate Next Part", variant="primary")
-                    generated_part = gr.Textbox(label="Generated Continuation", lines=5)
-        
-        with gr.TabItem("Summary"):
-            summarize_btn = gr.Button("Summarize Story", variant="primary")
-            summary = gr.Textbox(label="Story Summary", lines=8)
     
     submit_btn.click(add_contribution, inputs=[contributor, content], outputs=[status, current_story])
     search_btn.click(get_similar_parts, inputs=[search_query, num_results], outputs=[search_results])