Update app.py

app.py

@@ -1,31 +1,31 @@
-import gradio as gr
-from ScriptMatcher import ScriptMatcher
-# Initialize the ScriptMatcher instance
-scriptmatcher = ScriptMatcher()
-
-def classify_movie_genre(description, genres):
-    """
-    Given a description (synopsis) and genres, return similar series predictions.
-    """
-    # Split the genres string into a list of keywords
-    genre_keywords = genres.split(",")  # Assuming genres are comma-separated
-    # Get the predictions using the ScriptMatcher
-    predictions = scriptmatcher.find_similar_series(description, genre_keywords)
-    
-    return predictions
-
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=classify_movie_genre,
-    inputs=[
-        gr.Textbox(lines=5, label="Synopsis (Description)"),
-        gr.Textbox(label="Genres (Comma-separated)")
-    ],
-    outputs=gr.Dataframe(label="Similar Series Predictions"),
-    live=False,  # No need for live updates as the processing will be based on submission
-    title="Genre Prediction",
-    description="Provide a movie synopsis and genres to get predictions for similar scripts.",
-)
-
-# Launch the Gradio interface
+import gradio as gr
+from ScriptMatcher import ScriptMatcher
+# Initialize the ScriptMatcher instance
+scriptmatcher = ScriptMatcher()
+
+def classify_movie_genre(description, genres):
+    """
+    Given a description (synopsis) and genres, return similar series predictions.
+    """
+    # Split the genres string into a list of keywords
+    genre_keywords = genres.split(",")  # Assuming genres are comma-separated
+    # Get the predictions using the ScriptMatcher
+    predictions = scriptmatcher.find_similar_series(description, genre_keywords)
+    
+    return pd.DataFrame(predictions)
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=classify_movie_genre,
+    inputs=[
+        gr.Textbox(lines=5, label="Synopsis (Description)"),
+        gr.Textbox(label="Genres (Comma-separated)")
+    ],
+    outputs=gr.Dataframe(label="Similar Series Predictions"),
+    live=False,  # No need for live updates as the processing will be based on submission
+    title="Genre Prediction",
+    description="Provide a movie synopsis and genres to get predictions for similar scripts.",
+)
+
+# Launch the Gradio interface
 iface.launch(inline=False)