update upload function

app.py

@@ -29,17 +29,20 @@ def prediction(news):
 
 def file_prediction(file):
     if file.name.endswith('.csv'):
-        df = pd.read_csv(file)  
-        news_list = df.iloc[:, 0].tolist()  
+        df = pd.read_csv(file) 
+        news_list = df.iloc[:, 0].tolist() 
     else:
-        file.seek(0) 
-        news_list = file.read().decode('utf-8').splitlines() 
+        file.seek(0)  
+        file_content = file.read().decode('utf-8')  
+        news_list = file_content.splitlines()
     
     results = []
     for news in news_list:
-        results.append(prediction(news))
+        if news.strip():
+            results.append(prediction(news))
     
-    return results
+    return pd.DataFrame(results, index=news_list)  
+
 
 
 gradio_ui = gr.Interface(