Update app.py

working for 1 url

app.py

@@ -118,20 +118,14 @@ def classify_website(url):
 
     
     urls = [url]
-    print(urls)
     results_shop = main(urls)
 
     # Convert results to DataFrame
     df_result_train_more = pd.DataFrame(results_shop)
-    print(df_result_train_more)
     text = df_result_train_more['text'][0]
-    print(text)
-    try:
-        translated = GoogleTranslator(source='auto', target='en').translate(text[:4990])
-    # except:
-        
+    translated = GoogleTranslator(source='auto', target='en').translate(text[:4990])
 
-    # try:
+    try:
         # Prepare the input prompt for the model
         prompt = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
 ### Instruction:
@@ -161,38 +155,14 @@ Categorize the website into one of the 3 categories:
         logging.exception(e)
         return str(e)
 
-
-def classify_urls_from_csv(csv_file):
-    # Read CSV file and extract URLs from the first column
-    df = pd.read_csv(csv_file)
-    df.iloc[:, 0] = df.iloc[:, 0].str.replace(';', '').str.strip()
-
-    urls = df.iloc[:, 0].tolist()
-
-    # Classify each URL and store the results
-    predictions = []
-    for url in urls:
-        prediction = classify_website(url)
-        predictions.append(prediction)
-
-    # Add predictions as a new column in the dataframe
-    df['Prediction'] = predictions
-
-    # Save the results to a new CSV file
-    output_file = "predictions.csv"
-    df.to_csv(output_file, index=False)
-
-    return output_file
-
-
 # Create a Gradio interface
 iface = gr.Interface(
-    fn=classify_urls_from_csv,
-    inputs=gr.File(label="Upload CSV (First column must contain URLs)"),
-    outputs=gr.File(label="Download CSV with Predictions"),
+    fn=classify_website,
+    inputs="text",
+    outputs="text",
     title="Website Categorization",
-    description="Upload a CSV file with URLs in the first column and categorize each website into one of the 3 categories: OTHER, NEWS/BLOG, or E-commerce."
+    description="Categorize a website into one of the 3 categories: OTHER, NEWS/BLOG, or E-commerce."
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()