Update app.py

app.py

@@ -5,10 +5,11 @@ from datetime import datetime
 from pixeltable.functions.huggingface import sentence_transformer
 from pixeltable.functions import openai
 import os
+import getpass
 
-# Ensure OpenAI API key is set
+# Store API keys
 if 'OPENAI_API_KEY' not in os.environ:
-    os.environ['OPENAI_API_KEY'] = input('Enter your OpenAI API key: ')
+    os.environ['OPENAI_API_KEY'] = getpass.getpass('OpenAI API key:')
 
 # Initialize Pixeltable
 pxt.drop_dir('story_builder', force=True)
@@ -65,6 +66,13 @@ story_table['continuation_response'] = openai.chat_completions(
     max_tokens=50
 )
 
+# Function to get the current cumulative story
+def get_current_story():
+    latest_entry = story_table.tail(1)
+    if len(latest_entry) > 0:
+        return latest_entry['cumulative_story'][0]
+    return ""
+
 # Functions for Gradio interface
 def add_contribution(contributor, content):
     current_story = get_current_story()
@@ -80,7 +88,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).collect()
+    results = story_table.order_by(sim, asc=False).limit(num_results).select(story_table.content, story_table.contributor, sim=sim).collect()
     return results.to_pandas()
 
 def generate_next_part():
@@ -91,6 +99,7 @@ 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(
     """
@@ -114,30 +123,33 @@ 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 Nickname")
+                    contributor = gr.Textbox(label="Your Name")
                     content = gr.Textbox(label="Your Contribution", lines=5)
                     submit_btn = gr.Button("Submit Contribution", variant="primary")
-            
-            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.Column(scale=3):
+                    status = gr.Textbox(label="Status")
+                    current_story = gr.Textbox(label="Current Story", lines=10, interactive=False)
         
-        with gr.TabItem("Search Past Contributions"):
+        with gr.TabItem("Search & Generate"):
             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"],
+                        headers=["Content", "Contributor", "Similarity"],
                         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])
+    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])
     generate_btn.click(generate_next_part, outputs=[generated_part])
     summarize_btn.click(summarize_story, outputs=[summary])