update Readme

README.md

@@ -1,8 +1,8 @@
 ---
 title: VisualNovelLM
-emoji: ⚡
-colorFrom: red
-colorTo: red
+emoji: 🎮📖🤖
+colorFrom: pink
+colorTo: indigo
 sdk: gradio
 sdk_version: 4.27.0
 app_file: app.py

app.py

@@ -70,31 +70,6 @@ def invoke_lambda_function(story_prompt, lore_file_s3_path):
         return None
 
 
-# def invoke_lambda_function(story_prompt, lore_file_s3_path):
-#     payload = {"story_prompt": story_prompt, "lore_file": lore_file_s3_path}
-#     try:
-#         print("Invoking Lambda function...")
-#         response = lambda_client.invoke(
-#             FunctionName="renpy_builder",
-#             InvocationType="RequestResponse",
-#             Payload=json.dumps(payload),
-#         )
-#         print("Finished")
-
-#         # Parse the response
-#         response_payload = json.loads(response["Payload"].read())
-#         print("Lambda response:", response_payload)
-
-#         if response_payload.get("success", False):
-#             return response_payload["download_url"]
-#         else:
-#             raise Exception(response_payload.get("message", "Unknown error"))
-
-#     except Exception as e:
-#         print(f"Error invoking Lambda: {str(e)}")
-#         raise
-
-
 def load_existing_game(session_id):
     """Load an existing game by session ID"""
     if not session_id:
@@ -185,7 +160,9 @@ with gr.Blocks() as demo:
             placeholder="Describe the story you want to generate. Include details about the setting, main characters, and key plot points.",
             lines=5,
         )
-        lore_file = gr.File(label="Upload lore document (optional)")
+        lore_file = gr.File(
+            label="Upload lore document (optional)", file_types=["text"]
+        )
 
     # Add example prompts
     gr.Examples(
@@ -227,18 +204,18 @@ with gr.Blocks() as demo:
         fn=build_and_display_game, inputs=[story_prompt, lore_file], outputs=output
     )
 
-    with gr.Tab("Load Existing Game"):
-        gr.Markdown("Enter a session ID to load an existing game")
+    # with gr.Tab("Load Existing Game"):
+    #     gr.Markdown("Enter a session ID to load an existing game")
 
-        session_id_input = gr.Textbox(
-            label="Session ID", placeholder="Enter the session ID of an existing game"
-        )
-        load_button = gr.Button("Load Game")
-        load_output = gr.HTML(label="Game Output")
+    #     session_id_input = gr.Textbox(
+    #         label="Session ID", placeholder="Enter the session ID of an existing game"
+    #     )
+    #     load_button = gr.Button("Load Game")
+    #     load_output = gr.HTML(label="Game Output")
 
-        load_button.click(
-            fn=load_existing_game, inputs=session_id_input, outputs=load_output
-        )
+    #     load_button.click(
+    #         fn=load_existing_game, inputs=session_id_input, outputs=load_output
+    #     )
 
 # Create FastAPI app
 app = FastAPI()